Merge branch 'feature/mlflow-experiment-tracking' into feature/sc-77250-reload-mlflow-artifact-registry

Author: Valentin Thorey

dataikuapi/dss/project.py

@@ -1582,7 +1582,7 @@ def list_hive_tables(self, hive_database):
         """
         connection_name = "@virtual(hive-jdbc):" + hive_database
         ret = self.client._perform_json("GET", "/projects/%s/datasets/tables-import/actions/list-tables" % (self.project_key),
-                params = {"connectionName": connection_name} )
+                params={"connectionName": connection_name} )
 
         def to_schema_table_pair(x):
             return {"schema":x.get("databaseName", None), "table":x["table"]}
@@ -1591,13 +1591,11 @@ def to_schema_table_pair(x):
     ########################################################
     # App designer
     ########################################################
-
     def get_app_manifest(self):
         raw_data = self.client._perform_json("GET", "/projects/%s/app-manifest" % self.project_key)
         return DSSAppManifest(self.client, raw_data, self.project_key)
 
-    ########################################################
-    # MLflow
+    # MLflow experiment tracking
     ########################################################
     def setup_mlflow(self, managed_folder="mlflow_artifacts", host=None):
         """
@@ -1608,6 +1606,14 @@ def setup_mlflow(self, managed_folder="mlflow_artifacts", host=None):
         """
         return MLflowHandle(client=self.client, project_key=self.project_key, managed_folder=managed_folder, host=host)
 
+    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):
     """

setup.py

@@ -4,30 +4,31 @@
 
 VERSION = "9.0.0"
 
-long_description = (open('README').read() +
-    '\n\n' + open('HISTORY.txt').read())
+long_description = (open('README').read() + '\n\n' +
+                    open('HISTORY.txt').read())
 
 setup(
-        name='dataiku-api-client',
-        version=VERSION,
-        license="Apache Software License",
-        packages=["dataikuapi", "dataikuapi.dss", "dataikuapi.apinode_admin", "dataikuapi.fm"],
-        description="Python API client for Dataiku APIs",
-        long_description=long_description,
-        author="Dataiku",
-        author_email="support@dataiku.com",
-        url="https://www.dataiku.com",
-        download_url = "https://github.com/dataiku/dataiku-api-client-python/tarball/" + VERSION,
-        classifiers = [
-            'Development Status :: 5 - Production/Stable',
-            'Intended Audience :: Developers',
-            'License :: OSI Approved :: Apache Software License',
-            'Topic :: Software Development :: Libraries',
-            'Programming Language :: Python',
-            'Operating System :: OS Independent'
-        ],
-        install_requires = [
-            "requests>=2,<2.22",
-            "python-dateutil"
-        ]
-     )
+    name='dataiku-api-client',
+    version=VERSION,
+    license="Apache Software License",
+    packages=["dataikuapi", "dataikuapi.dss", "dataikuapi.apinode_admin", "dataikuapi.fm",
+              "dataikuapi.dss_plugin_mlflow"],
+    description="Python API client for Dataiku APIs",
+    long_description=long_description,
+    author="Dataiku",
+    author_email="support@dataiku.com",
+    url="https://www.dataiku.com",
+    download_url="https://github.com/dataiku/dataiku-api-client-python/tarball/" + VERSION,
+    classifiers=[
+        'Development Status :: 5 - Production/Stable',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: Apache Software License',
+        'Topic :: Software Development :: Libraries',
+        'Programming Language :: Python',
+        'Operating System :: OS Independent'
+    ],
+    install_requires=[
+        "requests>=2,<2.22",
+        "python-dateutil"
+    ]
+)