Skip to content

Commit 0740b68

Browse files
committed
Add xmltodict-based list-transformation to run traces
1 parent 6675209 commit 0740b68

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

openml/runs/functions.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -699,14 +699,17 @@ def _create_run_from_xml(xml):
699699

700700

701701
def _create_trace_from_description(xml):
702-
result_dict = xmltodict.parse(xml)['oml:trace']
702+
result_dict = xmltodict.parse(xml, force_list=('oml:trace_iteration',))['oml:trace']
703703

704704
run_id = result_dict['oml:run_id']
705705
trace = dict()
706706

707707
if 'oml:trace_iteration' not in result_dict:
708708
raise ValueError('Run does not contain valid trace. ')
709709

710+
assert type(result_dict['oml:trace_iteration']) == list, \
711+
type(result_dict['oml:trace_iteration'])
712+
710713
for itt in result_dict['oml:trace_iteration']:
711714
repeat = int(itt['oml:repeat'])
712715
fold = int(itt['oml:fold'])

0 commit comments

Comments
 (0)