File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ),
You can’t perform that action at this time.
0 commit comments