Skip to content

Commit 3aea218

Browse files
committed
different error catched
1 parent 265b034 commit 3aea218

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/test_runs/test_run_functions.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def test_run_regression_on_classif_task(self):
4141

4242
clf = LinearRegression()
4343
task = openml.tasks.get_task(task_id)
44-
self.assertRaises(openml.exceptions.PyOpenMLError, openml.runs.run_task,
44+
self.assertRaises(openml.exceptions.AttributeError, openml.runs.run_task,
4545
task=task, model=clf, avoid_duplicate_runs=False)
4646

4747
@mock.patch('openml.flows.sklearn_to_flow')
@@ -83,6 +83,7 @@ def test_run_optimize_randomforest_iris(self):
8383
n_iter=num_iterations)
8484

8585
run = self._perform_run(task_id, num_instances, random_search)
86+
print(run.trace_content)
8687
self.assertEqual(len(run.trace_content), num_iterations * num_folds)
8788

8889
def test_run_optimize_bagging_iris(self):

0 commit comments

Comments
 (0)