bugfixes

Author: janvanrijn

openml/evaluations/evaluation.py

@@ -2,7 +2,7 @@
 class OpenMLEvaluation(object):
 
     def __init__(self, run_id, task_id, setup_id, flow_id, flow_name,
-                 data_name, function, upload_time, value, array_data):
+                 data_name, function, upload_time, value, array_data=None):
         self.run_id = run_id
         self.task_id = task_id
         self.setup_id = setup_id

openml/evaluations/functions.py

@@ -6,7 +6,7 @@
 def list_evaluations(function, task_id):
     """Helper function to parse API calls which are lists of runs"""
 
-    xml_string = _perform_api_call("evaluation/list/function/%s/task_id/%d" %(function, task_id))
+    xml_string = _perform_api_call("evaluation/list/function/%s/task/%d" %(function, task_id))
 
     evals_dict = xmltodict.parse(xml_string)
     # Minimalistic check if the XML is useful
@@ -24,11 +24,15 @@ def list_evaluations(function, task_id):
     evals = dict()
     for eval_ in evals_list:
         run_id = int(eval_['oml:run_id'])
-        evaluation = OpenMLEvaluation(int(eval_['oml:run_id']), int(eval_['task_id']),
+        array_data = None
+        if 'oml:array_data' in eval_:
+            eval_['oml:array_data']
+
+        evaluation = OpenMLEvaluation(int(eval_['oml:run_id']), int(eval_['oml:task_id']),
                                       int(eval_['oml:setup_id']), int(eval_['oml:flow_id']),
                                       eval_['oml:flow_name'], eval_['oml:data_name'],
                                       eval_['oml:function'], eval_['oml:upload_time'],
-                                      float(eval_['oml:value']), eval_['oml:array_data'])
+                                      float(eval_['oml:value']), array_data)
         evals[run_id] = evaluation
-    return evaluation
+    return evals
 

tests/test_evaluations/test_evaluation_functions.py

@@ -8,7 +8,9 @@ class TestEvaluationFunctions(TestBase):
     def test_evaluation_list(self):
         openml.config.server = self.production_server
 
-        res = openml.evaluations.list_evaluations("predictive_accuracy", 59)
+        task_id = 7312
+
+        res = openml.evaluations.list_evaluations("predictive_accuracy", task_id)
 
         self.assertGreater(len(res), 100)