Python KedroMlflowConfig.from_dict Examples

Example #1

File: mlflow_context.py Project: felipeeeantunes/kedro-mlflow

def get_mlflow_config(context: KedroContext):
    try:
        conf_mlflow_yml = context.config_loader.get("mlflow*", "mlflow*/**")
    except MissingConfigException:
        raise KedroMlflowConfigError(
            "No 'mlflow.yml' config file found in environment. Use ``kedro mlflow init`` command in CLI to create a default config file."
        )
    conf_mlflow = KedroMlflowConfig(context.project_path)
    conf_mlflow.from_dict(conf_mlflow_yml)
    return conf_mlflow

Example #2

def get_mlflow_config(session: Optional[KedroSession] = None):
    session = session or get_current_session()
    context = session.load_context()
    try:
        conf_mlflow_yml = context.config_loader.get("mlflow*", "mlflow*/**")
    except MissingConfigException:
        raise KedroMlflowConfigError(
            "No 'mlflow.yml' config file found in environment. Use ``kedro mlflow init`` command in CLI to create a default config file."
        )
    conf_mlflow = KedroMlflowConfig(context.project_path)
    conf_mlflow.from_dict(conf_mlflow_yml)
    return conf_mlflow

Example #3

File: test_config.py Project: gabrielbckr/kedro-mlflow

def test_from_dict_to_dict_idempotent(mocker, kedro_project_with_mlflow_conf):
    mocker.patch("mlflow.tracking.MlflowClient", return_value=None)
    mocker.patch(
        "kedro_mlflow.framework.context.config.KedroMlflowConfig._get_or_create_experiment",
        return_value=None,
    )

    config = KedroMlflowConfig(project_path=kedro_project_with_mlflow_conf)
    original_config_dict = config.to_dict()
    # modify config
    config.from_dict(original_config_dict)
    assert config.to_dict() == original_config_dict

Example #4

File: mlflow_context.py Project: takikadiri/kedro-mlflow

def get_mlflow_config(project_path=None, env="local"):
    if project_path is None:
        project_path = Path.cwd()
    project_path = Path(project_path)
    conf_paths = [
        str(project_path / "conf" / "base"),
        str(project_path / "conf" / env),
    ]
    config_loader = ConfigLoader(conf_paths=conf_paths)
    conf_mlflow_yml = config_loader.get("mlflow*", "mlflow*/**")
    conf_mlflow = KedroMlflowConfig(project_path=project_path)
    conf_mlflow.from_dict(conf_mlflow_yml)
    return conf_mlflow

Example #5

def get_mlflow_config(context: KedroContext):

    conf_mlflow_yml = context.config_loader.get("mlflow*", "mlflow*/**")
    conf_mlflow = KedroMlflowConfig(context.project_path)
    conf_mlflow.from_dict(conf_mlflow_yml)
    return conf_mlflow