Skip to content

Commit 30840fd

Browse files
committed
fix for bug created when refactoring for issue #210
1 parent b4e052d commit 30840fd

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

openml/runs/functions.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,10 +180,11 @@ def _run_task_get_arffcontent(model, task, class_labels):
180180

181181
fold_no = fold_no + 1
182182
rep_no = rep_no + 1
183-
184-
if get_traceble_model(model):
183+
184+
traceable_model = get_traceble_model(model)
185+
if traceable_model:
185186
# arff_tracecontent is already set
186-
arff_trace_attributes = _extract_arfftrace_attributes(model_fold)
187+
arff_trace_attributes = _extract_arfftrace_attributes(traceable_model)
187188
else:
188189
arff_tracecontent = None
189190
arff_trace_attributes = None

0 commit comments

Comments
 (0)