06e9433720845ab9f12e7ea0d8a8982123f093c6,tests/tracking/fluent/test_fluent_autolog.py,,test_universal_autolog_calls_specific_autologs_correctly,#Any#Any#,82

Before Change

        inspect.Parameter(param, inspect.Parameter.POSITIONAL_OR_KEYWORD, annotation=type_)
        for param, type_ in args.items()
    ]
    with mock.patch(
        "mlflow." + mlflow_module + ".autolog", wraps=getattr(mlflow, mlflow_module).autolog
    ) as autolog_mock:
        autolog_mock.__signature__ = inspect.Signature(params)

        autolog_mock.assert_not_called()

        // this should attach import hooks to each library
        mlflow.autolog(log_input_examples=True, log_model_signatures=True, log_models=True)

        autolog_mock.assert_not_called()

        mlflow.utils.import_hooks.notify_module_loaded(library)

        // after each library is imported, its corresponding autolog function should have been called
        if library in integrations_with_config:
            autolog_mock.assert_called_once_with(
                log_input_examples=True, log_model_signatures=True, log_models=True
            )
        else:
            autolog_mock.assert_called_once_with(log_models=True)


@pytest.mark.large
def test_universal_autolog_calls_pyspark_immediately():
    library = pyspark
    mlflow_module = "spark"

After Change

        "log_models": False,
        "disable": True,
    }
    if library in integrations_with_additional_config:
        args_to_test.update({"log_input_examples": True, "log_model_signatures": True})

    mlflow.autolog(**args_to_test)
    mlflow.utils.import_hooks.notify_module_loaded(library)

    for arg_key, arg_value in args_to_test.items():

Instances