diff --git a/doc/progress.rst b/doc/progress.rst index ba6225986..b65df1926 100644 --- a/doc/progress.rst +++ b/doc/progress.rst @@ -8,6 +8,7 @@ Changelog 0.10.2 ~~~~~~ +* ADD #857: Adds task type ID to list_runs * DOC #862: Added license BSD 3-Clause to each of the source files. 0.10.1 diff --git a/openml/runs/functions.py b/openml/runs/functions.py index aefc2162a..9e7321d45 100644 --- a/openml/runs/functions.py +++ b/openml/runs/functions.py @@ -969,6 +969,7 @@ def __list_runs(api_call, output_format='dict'): 'setup_id': int(run_['oml:setup_id']), 'flow_id': int(run_['oml:flow_id']), 'uploader': int(run_['oml:uploader']), + 'task_type': int(run_['oml:task_type_id']), 'upload_time': str(run_['oml:upload_time']), 'error_message': str((run_['oml:error_message']) or '')} diff --git a/tests/test_runs/test_run_functions.py b/tests/test_runs/test_run_functions.py index d44a000d6..2773bc8d9 100644 --- a/tests/test_runs/test_run_functions.py +++ b/tests/test_runs/test_run_functions.py @@ -1128,7 +1128,7 @@ def _check_run(self, run): # error_message and run_details exist, too, but are not used so far. We need to update # this check once they are used! self.assertIsInstance(run, dict) - assert len(run) == 7, str(run) + assert len(run) == 8, str(run) def test_get_runs_list(self): # TODO: comes from live, no such lists on test