Merge branch 'develop' into feature_repr

Author: janvanrijn

openml/datasets/dataset.py

@@ -373,13 +373,17 @@ def get_features_by_type(self, data_type, exclude=None,
         result : list
             a list of indices that have the specified data type
         '''
-        assert data_type in OpenMLDataFeature.LEGAL_DATA_TYPES, "Illegal feature type requested"
+        if data_type not in OpenMLDataFeature.LEGAL_DATA_TYPES:
+            raise TypeError("Illegal feature type requested")
         if self.ignore_attributes is not None:
-            assert type(self.ignore_attributes) is list, "ignore_attributes should be a list"
+            if not isinstance(self.ignore_attributes, list):
+                raise TypeError("ignore_attributes should be a list")
         if self.row_id_attribute is not None:
-            assert type(self.row_id_attribute) is str, "row id attribute should be a str"
+            if not isinstance(self.row_id_attribute, six.string_types):
+                raise TypeError("row id attribute should be a str")
         if exclude is not None:
-            assert type(exclude) is list, "Exclude should be a list"
+            if not isinstance(exclude, list):
+                raise TypeError("Exclude should be a list")
             # assert all(isinstance(elem, str) for elem in exclude), "Exclude should be a list of strings"
         to_exclude = []
         if exclude is not None:

openml/tasks/functions.py

@@ -142,7 +142,7 @@ def _list_tasks(api_call):
         xml_string = _perform_api_call(api_call)
     except OpenMLServerNoResult:
         return []
-    tasks_dict = xmltodict.parse(xml_string, force_list=('oml:task',))
+    tasks_dict = xmltodict.parse(xml_string, force_list=('oml:task','oml:input'))
     # Minimalistic check if the XML is useful
     if 'oml:tasks' not in tasks_dict:
         raise ValueError('Error in return XML, does not contain "oml:runs": %s'

tests/test_tasks/test_task_functions.py

@@ -42,6 +42,12 @@ def test__get_estimation_procedure_list(self):
         self.assertIsInstance(estimation_procedures[0], dict)
         self.assertEqual(estimation_procedures[0]['task_type_id'], 1)
 
+    def test_list_clustering_task(self):
+        # as shown by #383, clustering tasks can give list/dict casting problems
+        openml.config.server = self.production_server
+        openml.tasks.list_tasks(task_type_id=5, size=10)
+        # the expected outcome is that it doesn't crash. No assertions.
+
     def _check_task(self, task):
         self.assertEqual(type(task), dict)
         self.assertGreaterEqual(len(task), 2)
@@ -127,7 +133,6 @@ def assert_and_raise(*args, **kwargs):
             os.path.join(os.getcwd(), "tasks", "1", "tasks.xml")
         ))
 
-
     def test_get_task_with_cache(self):
         openml.config.set_cache_directory(self.static_cache_dir)
         task = openml.tasks.get_task(1)