comment wording + add get_raw to DSSMLTaskQueues

Author: exil

dataikuapi/dss/ml.py

@@ -3427,10 +3427,13 @@ def guess(self, prediction_type=None, reguess_level=None):
 
 class DSSMLTaskQueues(object):
     """
-    Object containing a list of MLTask queues
+    Iterable listing of MLTask queues
     """
-    def __init__(self, queues):
-        self.queues = queues
+    def __init__(self, data):
+        self.data = data
 
     def __iter__(self):
-        return self.queues.__iter__()
+        return self.data["queues"].__iter__()
+
+    def get_raw(self):
+        return self.data

dataikuapi/dss/project.py

@@ -593,13 +593,13 @@ def get_ml_task(self, analysis_id, mltask_id):
 
     def list_mltask_queues(self):
         """
-        List ML task queues in this project 
+        List non-empty ML task queues in this project 
         
-        :returns: a :class:`DSSMLTaskQueues` object containing a list of queues as dicts
+        :returns: an iterable :class:`DSSMLTaskQueues` listing of MLTask queues (each a dict)
         :rtype: :class:`DSSMLTaskQueues`
         """ 
-        ref = self.client._perform_json("GET", "/projects/%s/models/labs/mltask-queues" % self.project_key)
-        return DSSMLTaskQueues(ref["queues"])
+        data = self.client._perform_json("GET", "/projects/%s/models/labs/mltask-queues" % self.project_key)
+        return DSSMLTaskQueues(data)
 
     def create_analysis(self, input_dataset):
         """