Move clean-db with other mlflow extensions.

Author: lpenet

dataikuapi/dss/mlflow.py

@@ -92,3 +92,12 @@ def garbage_collect(self):
             "GET", "/api/2.0/mlflow/extension/garbage-collect",
             headers={"x-dku-mlflow-project-key": self.project_key}
         )
+
+    def clean_experiment_tracking_db(self):
+        """
+        Cleans the experiments, runs, params, metrics, tags, etc. for this project
+
+        This call requires an API key with admin rights
+        """
+        self.client._perform_raw("DELETE", "/api/2.0/mlflow/extension/clean-db/%s" % self.project_key)
+

dataikuapi/dss/project.py

@@ -1626,15 +1626,6 @@ def get_mlflow_extension(self):
         """
         return DSSMLflowExtension(client=self.client, project_key=self.project_key)
 
-    def clean_experiment_tracking_db(self):
-        """
-        Cleans the experiments, runs, params, metrics, tags, etc. for this project
-
-        This call requires an API key with admin rights
-        """
-        self.client._perform_raw("DELETE", "/api/2.0/mlflow/clean-db/%s" % self.project_key)
-
-
 class TablesImportDefinition(object):
     """
     Temporary structure holding the list of tables to import