Configuration
- class kedro_mlflow.config.kedro_mlflow_config.DictParamsOptions(*, flatten: StrictBool = False, recursive: StrictBool = True, sep: str = '.')
Bases:
BaseModel
- flatten: StrictBool
- recursive: StrictBool
- sep: str
- class kedro_mlflow.config.kedro_mlflow_config.DisableTrackingOptions(*, pipelines: List[str] = [])
Bases:
BaseModel
- pipelines: List[str]
- class kedro_mlflow.config.kedro_mlflow_config.ExperimentOptions(*, name: str = 'Default', restore_if_deleted: StrictBool = True)
Bases:
BaseModel
- 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
- server: MlflowServerOptions
- setup(context)
Setup all the mlflow configuration
- tracking: MlflowTrackingOptions
- 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
- 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
- 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
- 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
- init_kwargs: Dict[str, str]
- pass_context: bool
- type: str | None