Skip to content

Commit bcd9b77

Browse files
authored
Add clean_experiment_tracking_db to DSSProject (#199)
1 parent c670853 commit bcd9b77

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

dataikuapi/dss/project.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1595,6 +1595,17 @@ def get_app_manifest(self):
15951595
raw_data = self.client._perform_json("GET", "/projects/%s/app-manifest" % self.project_key)
15961596
return DSSAppManifest(self.client, raw_data, self.project_key)
15971597

1598+
########################################################
1599+
# MLflow experiment tracking
1600+
########################################################
1601+
def clean_experiment_tracking_db(self):
1602+
"""
1603+
Cleans the experiments, runs, params, metrics, tags, etc. for this project
1604+
1605+
This call requires an API key with admin rights
1606+
"""
1607+
self.client._perform_raw("DELETE", "/api/2.0/mlflow/clean-db/%s" % self.project_key)
1608+
15981609

15991610
class TablesImportDefinition(object):
16001611
"""

0 commit comments

Comments
 (0)