Experiment for MLFlow models

Author: cstenac

dataikuapi/dss/project.py

@@ -652,6 +652,21 @@ def get_saved_model(self, sm_id):
         """
         return DSSSavedModel(self.client, self.project_key, sm_id)
 
+    def create_mlflow_pyfunc_model(self, id):
+        """Starts creation of a new external saved model for storing 3rd party models
+
+        :param string id: 
+        """
+        if len(id) != 8:
+            raise ValueError("model id must be 8 characters long")
+        model = {
+            "projectKey" : self.project_key,
+            "id": id,
+            "savedModelType" : "MLFLOW_PYFUNC"
+        }
+        self.client._perform_empty("POST", "/projects/%s/savedmodels/" % self.project_key, body = model)
+        return self.get_saved_model(id)
+
     ########################################################
     # Managed folders
     ########################################################

dataikuapi/dss/savedmodel.py

@@ -117,6 +117,13 @@ def get_origin_ml_task(self):
         if fmi is not None:
             return DSSMLTask.from_full_model_id(self.client, fmi, project_key=self.project_key)
 
+    def import_version_from_folder(self, version_id, folder):
+        import shutil
+        shutil.make_archive("tmpmodel", "zip", folder) #[, root_dir[, base_dir[, verbose[, dry_run[, owner[, group[, logger]]]]]]])
+        
+        with open("tmpmodel.zip", "rb") as fp:
+            self.client._perform_empty("POST", "/projects/%s/savedmodels/%s/versions/%s" % (self.project_key, self.sm_id, version_id),
+                files={"file":("tmpmodel.zip", fp)})
 
     ########################################################
     # Metrics