Converting to the "pass params as JSON body" Dataiku way.

Author: lpenet

dataikuapi/dss/savedmodel.py

@@ -1,10 +1,8 @@
-from ..utils import DataikuException
-from ..utils import DataikuUTF8CSVReader
-from ..utils import DataikuStreamedHttpUTF8CSVReader
-import json
 from .ml import DSSTrainedPredictionModelDetails, DSSTrainedClusteringModelDetails
 from .metrics import ComputedMetrics
-from .discussion import DSSObjectDiscussions
+from .ml import DSSTrainedClusteringModelDetails
+from .ml import DSSTrainedPredictionModelDetails
+
 
 class DSSSavedModel(object):
     """
@@ -79,12 +77,13 @@ def delete_versions(self, versions, remove_intermediate=True):
         model, an intermediate version is created every time a partition has finished training.
         :type remove_intermediate: bool
         """
+        body = {
+            "versions": versions,
+            "removeIntermediate": remove_intermediate
+        }
         self.client._perform_empty(
             "POST", "/projects/%s/savedmodels/%s/actions/delete-versions" % (self.project_key, self.sm_id),
-            params={
-                'versions': versions,
-                'removeIntermediate': remove_intermediate
-            })
+            body=body)
     ########################################################
     # Metrics
     ########################################################