1  """
 2  The ``mlflow.tracking`` module provides a Python CRUD interface to MLflow experiments
 3  and runs. This is a lower level API that directly translates to MLflow
 4  `REST API <../rest-api.html>`_ calls.
 5  For a higher level API for managing an "active run", use the :py:mod:`mlflow` module.
 6  """
 7  
 8  # Minimum APIs required for core tracing functionality of mlflow-tracing package.
 9  from mlflow.tracking._tracking_service.utils import (
10      _get_artifact_repo,
11      _get_store,
12      get_tracking_uri,
13      is_tracking_uri_set,
14      set_tracking_uri,
15  )
16  from mlflow.version import IS_TRACING_SDK_ONLY
17  
18  __all__ = [
19      "get_tracking_uri",
20      "set_tracking_uri",
21      "is_tracking_uri_set",
22      "_get_artifact_repo",
23      "_get_store",
24  ]
25  
26  # Importing the following APIs only if mlflow or mlflow-skinny is installed.
27  if not IS_TRACING_SDK_ONLY:
28      from mlflow.tracking._model_registry.utils import (
29          get_registry_uri,
30          set_registry_uri,
31      )
32      from mlflow.tracking._tracking_service.utils import _get_artifact_repo
33      from mlflow.tracking.client import MlflowClient
34  
35      __all__ += [
36          "get_registry_uri",
37          "set_registry_uri",
38          "MlflowClient",
39      ]