Skip to content

Commit c80c1b0

Browse files
committed
made sure we can assume tasp inputs is a list #383
1 parent ef955d5 commit c80c1b0

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

openml/tasks/functions.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,11 @@ def _list_tasks(api_call):
176176
'status': task_['oml:status']}
177177

178178
# Other task inputs
179-
for input in task_.get('oml:input', list()):
179+
task_inputs = task_.get('oml:input')
180+
if isinstance(task_inputs, dict):
181+
task_inputs = [task_inputs]
182+
183+
for input in task_inputs:
180184
if input['@name'] == 'estimation_procedure':
181185
task[input['@name']] = proc_dict[int(input['#text'])]['name']
182186
else:

0 commit comments

Comments
 (0)