Skip to content

Commit 5234d42

Browse files
committed
Merge pull request #95 from openml/feature/download_runs
FIX key error in creation of run object
2 parents 72acf39 + 41cae4f commit 5234d42

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

openml/runs/run.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -324,9 +324,9 @@ def _create_run_from_xml(xml):
324324
raise ValueError('Could not find keys "value" or "array_data" '
325325
'in %s' % str(evaluation_dict.keys()))
326326

327-
if 'oml:repeat' in evaluation_dict:
328-
repeat = int(evaluation_dict['oml:repeat'])
329-
fold = int(evaluation_dict['oml:fold'])
327+
if '@repeat' in evaluation_dict and '@fold' in evaluation_dict:
328+
repeat = int(evaluation_dict['@repeat'])
329+
fold = int(evaluation_dict['@fold'])
330330
repeat_dict = detailed_evaluations[key]
331331
fold_dict = repeat_dict[repeat]
332332
fold_dict[fold] = value

0 commit comments

Comments
 (0)