Adding python model versioning capabilities and fix python wrapper score handle null/none inputs

Author: Cloud User

src/sasctl/_services/model_repository.py

@@ -7,7 +7,7 @@
 """The Model Repository service supports registering and managing models."""
 
 from .service import Service
-from ..core import current_session, get
+from ..core import current_session, get, delete
 
 FUNCTIONS = {'Analytical', 'Classification', 'Clustering', 'Forecasting',
              'Prediction', 'Text categorization', 'Text extraction',
@@ -497,3 +497,24 @@ def copy_analytic_store(cls, model):
             model = cls.get_model(model, refresh=True)
 
         return cls.request_link(model, 'copyAnalyticStore')
+
+
+    @classmethod
+    def delete_model_contents(cls, model):
+        """Deletes all contents (files) in the model.
+
+        Parameters
+        ----------
+        model : str or dict
+            The name, id, or dictionary representation of a model.
+
+        Returns
+        -------
+
+        """
+        rel = 'delete'
+
+        filelist=cls.get_model_contents(model)
+        for delfile in filelist:
+            modelfileuri=cls.get_link(delfile, rel)
+            delete(modelfileuri['uri'])

src/sasctl/tasks.py

@@ -18,7 +18,7 @@
 from six.moves.urllib.error import HTTPError
 
 from . import utils
-from .core import RestObj, current_session, get, get_link, request_link
+from .core import RestObj, current_session, get, get_link, request_link, delete
 from .exceptions import AuthorizationError
 from .services import model_management as mm
 from .services import model_publish as mp
@@ -136,41 +136,6 @@ def register_model(model, name, project, repository=None, input=None,
     # If version not specified, default to creating a new version
     version = version or 'new'
 
-    # If replacing an existing version, make sure the model version exists
-    if str(version).lower() != 'new':
-        model_obj = mr.get_model(name)
-        if model_obj is None:
-            raise ValueError("Unable to update version '%s' of model '%s%.  "
-                             "Model not found." % (version, name))
-        model_versions = request_link(model_obj, 'modelVersions')
-        assert isinstance(model_versions, list)
-
-        # Use 'new' to create a new version if one doesn't exist yet.
-        if len(model_versions) == 0:
-            raise ValueError("No existing version of model '%s' to update."
-                             % name)
-
-        # Help function for extracting version number of REST response
-        def get_version(x):
-            return float(x.get('modelVersionName', 0))
-
-        if str(version).isnumeric():
-            match = [x for x in model_versions if float(version) ==
-                     get_version(x)]
-            assert len(match) <= 1
-
-            match = match[0] if len(match) else None
-        elif str(version).lower() == 'latest':
-            # Sort by version number and get first
-            match = sorted(model_versions, key=get_version)[0]
-        else:
-            raise ValueError("Unrecognized version '%s'." % version)
-
-        # TODO: get ID of correct model version
-    # if version != new, get existing model
-    # get model (modelVersions) rel
-    # -> returns list w/ id, modelVersionName, etc
-
     files = files or []
 
     # Find the project if it already exists
@@ -340,7 +305,18 @@ def get_version(x):
                 project['eventProbabilityVariable'] = prediction_variable
                 mr.update_project(project)
 
-    model = mr.create_model(model, project)
+    # If replacing an existing version, make sure the model version exists
+    if str(version).lower() != 'new':
+        #Update an existing model with new files
+        model_obj = mr.get_model(name)
+        if model_obj is None:
+            raise ValueError("Unable to update version '%s' of model '%s%.  "
+                             "Model not found." % (version, name))
+        model = mr.create_model_version(name)
+        mr.delete_model_contents(model)
+    else:
+        #Assume new model to create
+        model = mr.create_model(model, project)
 
     assert isinstance(model, RestObj)