adding resume queue and list queue calls

Author: exil

dataikuapi/dss/ml.py

@@ -3328,6 +3328,16 @@ def delete_trained_model(self, model_id):
         self.client._perform_empty(
             "DELETE", "/projects/%s/models/lab/%s/%s/models/%s" % (self.project_key, self.analysis_id, self.mltask_id, model_id))
 
+    def resume_queue(self):
+        """
+        Resumes a paused queue
+
+        :return: A dict including the sessionID of the resumed queue
+        :rtype dict
+        """
+        return self.client._perform_json(
+            "POST", "/projects/%s/models/lab/%s/%s/actions/resumeQueue" % (self.project_key, self.analysis_id, self.mltask_id))
+
     def deploy_to_flow(self, model_id, model_name, train_dataset, test_dataset=None, redo_optimization=True):
         """
         Deploys a trained model from this ML Task to a saved model + train recipe in the Flow.

dataikuapi/dss/project.py

@@ -591,6 +591,14 @@ def get_ml_task(self, analysis_id, mltask_id):
         """
         return DSSMLTask(self.client, self.project_key, analysis_id, mltask_id)
 
+    def list_mltask_queues(self):
+        """
+        List all paused ML task queues in this project
+        
+        Returns:
+            the list of the paused ML task queues, each one as a JSON object
+        """
+        return self.client._perform_json("GET", "/projects/%s/models/lab/mltask-queues" % self.project_key)
 
     def create_analysis(self, input_dataset):
         """