Coverage for src/mlopus/mlflow/__init__.py: 100%

1"""This module is based on the interface :class:`~mlopus.mlflow.BaseMlflowApi`, 

2which may be implemented in order to work with different MLflow backends/providers 

3in the scope of experiment tracking and model registry. 

4 

5Built-in implementations can be found under the module :mod:`mlopus.mlflow.providers` 

6and are also available under the plugin group `mlopus.mlflow_api_providers`. 

7 

8Third-party implementations may also be added to that group in order to expand funcionality. 

9""" 

10 

11from . import providers 

12from .api.common import schema, exceptions 

13from .api.base import BaseMlflowApi 

14from .api.run import RunApi 

15from .api.exp import ExpApi 

16from .api.model import ModelApi 

17from .api.mv import ModelVersionApi 

18from .utils import get_api, list_api_plugins, api_conf_schema 

19from .traits import MlflowRunMixin, MlflowApiMixin, MlflowRunManager 

20 

21RunStatus = schema.RunStatus 

22 

23__all__ = [ 

24 "get_api", 

25 "list_api_plugins", 

26 "api_conf_schema", 

27 "RunStatus", 

28 "exceptions", 

29 "providers", 

30 "MlflowRunMixin", 

31 "MlflowApiMixin", 

32 "MlflowRunManager", 

33 "BaseMlflowApi", 

34 "ExpApi", 

35 "RunApi", 

36 "ModelApi", 

37 "ModelVersionApi", 

38]