Skip to content

Commit 41cae4f

Browse files
committed
FIX key error in creation of run object
1 parent fc70dd3 commit 41cae4f

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
@@ -326,9 +326,9 @@ def _create_run_from_xml(xml):
326326
raise ValueError('Could not find keys "value" or "array_data" '
327327
'in %s' % str(evaluation_dict.keys()))
328328

329-
if 'oml:repeat' in evaluation_dict:
330-
repeat = int(evaluation_dict['oml:repeat'])
331-
fold = int(evaluation_dict['oml:fold'])
329+
if '@repeat' in evaluation_dict and '@fold' in evaluation_dict:
330+
repeat = int(evaluation_dict['@repeat'])
331+
fold = int(evaluation_dict['@fold'])
332332
repeat_dict = detailed_evaluations[key]
333333
fold_dict = repeat_dict[repeat]
334334
fold_dict[fold] = value

0 commit comments

Comments
 (0)