Skip to content

Commit 85e81d3

Browse files
committed
Adding delete_versions to the saved model API.
1 parent 5aaeee0 commit 85e81d3

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

dataikuapi/dss/savedmodel.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,22 @@ def set_active_version(self, version_id):
6969
self.client._perform_empty(
7070
"POST", "/projects/%s/savedmodels/%s/versions/%s/actions/setActive" % (self.project_key, self.sm_id, version_id))
7171

72+
def delete_versions(self, versions, remove_intermediate=True):
73+
"""
74+
Delete version(s) of the saved model
75+
76+
:param versions: list of versions to delete
77+
:type versions: list[str]
78+
:param remove_intermediate: also remove intermediate versions (default: True). In the case of a partitioned
79+
model, an intermediate version is created every time a partition has finished training.
80+
:type remove_intermediate: bool
81+
"""
82+
self.client._perform_empty(
83+
"POST", "/projects/%s/savedmodels/%s/actions/delete-versions" % (self.project_key, self.sm_id),
84+
params={
85+
'versions': json.dumps(versions),
86+
'removeIntermediate': remove_intermediate
87+
})
7288
########################################################
7389
# Metrics
7490
########################################################

0 commit comments

Comments
 (0)