Skip to content

Commit e5cdbe5

Browse files
committed
Fixing exception handling in new unit test
1 parent 2392630 commit e5cdbe5

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/test_datasets/test_dataset_functions.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1123,12 +1123,10 @@ def test_publish_fetch_ignore_attribute(self):
11231123
dataset = openml.datasets.get_dataset(upload_did)
11241124
break
11251125
except Exception as e:
1126+
# returned code 273: Dataset not processed yet
1127+
# returned code 362: No qualities found
11261128
trials += 1
1127-
if str(e).split(':')[-1].strip() == "Dataset not processed yet":
1128-
# if returned code 273: Dataset not processed yet
1129-
continue
1130-
else:
1131-
raise RuntimeError(str(e))
1129+
continue
11321130
if dataset is None:
11331131
raise ValueError("TIMEOUT: Failed to fetch uploaded dataset - {}".format(upload_did))
11341132
self.assertEqual(dataset.ignore_attribute, ignore_attribute)

0 commit comments

Comments
 (0)