Add DSSSavedModelSettings class

Author: Louis Pouillot

dataikuapi/dss/savedmodel.py

@@ -20,9 +20,16 @@ def __init__(self, client, project_key, sm_id):
     def id(self):
         return self.sm_id
 
-    def get_definition(self):
-        return self.client._perform_json(
+    def get_settings(self):
+        """
+        Returns the settings of this saved model.
+
+        :rtype: DSSSavedModelSettings
+        """
+        data = self.client._perform_json(
             "GET", "/projects/%s/savedmodels/%s" % (self.project_key, self.sm_id))
+        return DSSSavedModelSettings(self, data)
+
 
     ########################################################
     # Versions
@@ -102,10 +109,9 @@ def get_origin_ml_task(self):
 
         :rtype: DSSMLTask | None
         """
-        fmi = self.get_definition().get("lastExportedFrom")
+        fmi = self.get_settings().get_raw().get("lastExportedFrom")
         if fmi is not None:
-            origin_ml_task = DSSMLTask.from_full_model_id(self.client, fmi, project_key=self.project_key)
-            return origin_ml_task.get_trained_model_details(fmi)
+            return DSSMLTask.from_full_model_id(self.client, fmi, project_key=self.project_key)
 
 
     ########################################################
@@ -173,3 +179,13 @@ def delete(self):
 
         """
         return self.client._perform_empty("DELETE", "/projects/%s/savedmodels/%s" % (self.project_key, self.sm_id))
+
+
+class DSSSavedModelSettings:
+
+    def __init__(self, saved_model, settings):
+        self.saved_model = saved_model
+        self.settings = settings
+
+    def get_raw(self):
+        return self.settings