Dataiku’s experiment tracking features extensions to the MLflow Python API.

Some of those extensions are allowing actions that can only be performed through the CLI in standard MLflow.

In order to interact with these extensions, you must first obtain a reference to the dataikuapi.dss.mlflow.DSSMLflowExtension through dataikuapi.dss.project.DSSProject.get_mlflow_extension()

import dataiku
import mlflow

project = dataiku.api_client().get_default_project()
mlflow_extension = project.get_mlflow_extension()

You can then use the following methods:

Others are more DSS specific: