Merge pull request #206 from openml/fix205

if else around non guaranteed run element
openml · Mar 3, 2017 · c1c59d9 · c1c59d9
2 parents 93badda + d925655
commit c1c59d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/openml/runs/__init__.py b/openml/runs/__init__.py
@@ -1,4 +1,4 @@
 from .run import OpenMLRun
-from .functions import (run_task, get_run, list_runs)
+from .functions import (run_task, get_run, list_runs, get_runs)
 
-__all__ = ['OpenMLRun', 'run_task', 'get_run', 'list_runs']
+__all__ = ['OpenMLRun', 'run_task', 'get_run', 'list_runs', 'get_runs']
diff --git a/openml/runs/functions.py b/openml/runs/functions.py
@@ -201,7 +201,11 @@ def _create_run_from_xml(xml):
     uploader_name = run['oml:uploader_name']
     task_id = int(run['oml:task_id'])
     task_type = run['oml:task_type']
-    task_evaluation_measure = run['oml:task_evaluation_measure']
+    if 'oml:task_evaluation_measure' in run:
+        task_evaluation_measure = run['oml:task_evaluation_measure']
+    else:
+        task_evaluation_measure = None
+
     flow_id = int(run['oml:flow_id'])
     flow_name = run['oml:flow_name']
     setup_id = int(run['oml:setup_id'])