Configuration

class kedro_mlflow.config.kedro_mlflow_config.DictParamsOptions(*, flatten: StrictBool = False, recursive: StrictBool = True, sep: str = '.')

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

flatten: StrictBool

recursive: StrictBool

sep: str

class kedro_mlflow.config.kedro_mlflow_config.DisableTrackingOptions(*, pipelines: List[str] = [])

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

pipelines: List[str]

class kedro_mlflow.config.kedro_mlflow_config.ExperimentOptions(*, name: str = 'Default', restore_if_deleted: StrictBool = True)

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

name: str

restore_if_deleted: StrictBool

class kedro_mlflow.config.kedro_mlflow_config.KedroMlflowConfig(*, server: MlflowServerOptions = MlflowServerOptions(mlflow_tracking_uri=None, mlflow_registry_uri=None, credentials=None, request_header_provider=RequestHeaderProviderOptions(type=None, pass_context=False, init_kwargs={})), tracking: MlflowTrackingOptions = MlflowTrackingOptions(disable_tracking=DisableTrackingOptions(pipelines=[]), experiment=ExperimentOptions(name='Default', restore_if_deleted=True), run=RunOptions(id=None, name=None, nested=True), params=MlflowParamsOptions(dict_params=DictParamsOptions(flatten=False, recursive=True, sep='.'), long_params_strategy='fail')), ui: UiOptions = UiOptions(port='5000', host='127.0.0.1'))

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

validate_assignment = True

server: MlflowServerOptions

setup(context)

Setup all the mlflow configuration

tracking: MlflowTrackingOptions

ui: UiOptions

class kedro_mlflow.config.kedro_mlflow_config.MlflowParamsOptions(*, dict_params: DictParamsOptions = DictParamsOptions(flatten=False, recursive=True, sep='.'), long_params_strategy: Literal['fail', 'truncate', 'tag'] = 'fail')

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

dict_params: DictParamsOptions

long_params_strategy: Literal['fail', 'truncate', 'tag']

class kedro_mlflow.config.kedro_mlflow_config.MlflowServerOptions(*, mlflow_tracking_uri: str | None = None, mlflow_registry_uri: str | None = None, credentials: str | None = None, request_header_provider: RequestHeaderProviderOptions = RequestHeaderProviderOptions(type=None, pass_context=False, init_kwargs={}))

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

credentials: str | None

mlflow_registry_uri: str | None

mlflow_tracking_uri: str | None

request_header_provider: RequestHeaderProviderOptions

class kedro_mlflow.config.kedro_mlflow_config.MlflowTrackingOptions(*, disable_tracking: DisableTrackingOptions = DisableTrackingOptions(pipelines=[]), experiment: ExperimentOptions = ExperimentOptions(name='Default', restore_if_deleted=True), run: RunOptions = RunOptions(id=None, name=None, nested=True), params: MlflowParamsOptions = MlflowParamsOptions(dict_params=DictParamsOptions(flatten=False, recursive=True, sep='.'), long_params_strategy='fail'))

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

disable_tracking: DisableTrackingOptions

experiment: ExperimentOptions

params: MlflowParamsOptions

run: RunOptions

class kedro_mlflow.config.kedro_mlflow_config.RequestHeaderProviderOptions(*, type: str | None = None, pass_context: bool = False, init_kwargs: Dict[str, str] = {})

Bases: BaseModel

class Config

Bases: object

arbitrary_types_allowed = 'allowed'

extra = 'forbid'

init_kwargs: Dict[str, str]

pass_context: bool

type: str | None

class kedro_mlflow.config.kedro_mlflow_config.RunOptions(*, id: str | None = None, name: str | None = None, nested: StrictBool = True)

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

id: str | None

name: str | None

nested: StrictBool

class kedro_mlflow.config.kedro_mlflow_config.UiOptions(*, port: str = '5000', host: str = '127.0.0.1')

Bases: BaseModel

class Config

Bases: object

extra = 'forbid'

host: str

port: str