Skip to content

Commit 50f33c1

Browse files
authored
Merge branch 'develop' into speedup_travis
2 parents 3de0daf + afa36a1 commit 50f33c1

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

tests/test_runs/test_run_functions.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -498,7 +498,10 @@ def test_get_run_trace(self):
498498
# in case the run did not exists yet
499499
run = openml.runs.run_model_on_task(task, clf, avoid_duplicate_runs=True)
500500
trace = openml.runs.functions._create_trace_from_arff(run._generate_trace_arff_dict())
501-
self.assertEquals(len(trace['data']), num_iterations * num_folds)
501+
self.assertEquals(
502+
len(trace.trace_iterations),
503+
num_iterations * num_folds,
504+
)
502505
run = run.publish()
503506
self._wait_for_processed_run(run.run_id, 200)
504507
run_id = run.run_id
@@ -744,11 +747,10 @@ def test__create_trace_from_arff(self):
744747
trace_arff = arff.load(arff_file)
745748
trace = openml.runs.functions._create_trace_from_arff(trace_arff)
746749

747-
748750
def test_get_run(self):
749751
# this run is not available on test
750752
openml.config.server = self.production_server
751-
run = openml.runs.get_run(473350)
753+
run = openml.runs.get_run(473344)
752754
self.assertEqual(run.dataset_id, 1167)
753755
self.assertEqual(run.evaluations['f_measure'], 0.624668)
754756
for i, value in [(0, 0.66233),

0 commit comments

Comments
 (0)