Skip to content
Snippets Groups Projects
Commit 45043ddb authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

[test] Remove unnecessary padchest test code

parent fb46d403
No related branches found
No related tags found
2 merge requests!18Update tests,!16Make square centre-padding a model transform
...@@ -60,18 +60,7 @@ def test_loading(database_checkers, name: str, dataset: str, num_labels: int): ...@@ -60,18 +60,7 @@ def test_loading(database_checkers, name: str, dataset: str, num_labels: int):
datamodule.setup("predict") # sets up all datasets datamodule.setup("predict") # sets up all datasets
if dataset in datamodule.predict_dataloader(): if dataset in datamodule.predict_dataloader():
# Not all datamodules have a test set loader = datamodule.predict_dataloader()[dataset]
try:
loader = datamodule.predict_dataloader()[dataset]
except KeyError as e:
if str(e) == "test" and name in [
"idiap",
"no_tb_idiap",
"cardiomegaly_idiap",
]:
assert True
else:
assert False
limit = 3 # limit load checking limit = 3 # limit load checking
for batch in loader: for batch in loader:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment