def test_mlflow_model_logger_logging_deactivation_is_bool():
mlflow_model_logger_dataset = MlflowModelLoggerDataSet(
flavor="mlflow.sklearn")
with pytest.raises(ValueError,
match="_logging_activated must be a boolean"):
mlflow_model_logger_dataset._logging_activated = "hello"
def test_mlflow_model_logger_logging_deactivation(tracking_uri, linreg_model):
mlflow_model_logger_dataset = MlflowModelLoggerDataSet(
flavor="mlflow.sklearn")
mlflow.set_tracking_uri(tracking_uri)
mlflow_client = MlflowClient(tracking_uri=tracking_uri)
mlflow_model_logger_dataset._logging_activated = False
all_runs_id_beginning = set([
run.run_id for k in range(len(mlflow_client.list_experiments()))
for run in mlflow_client.list_run_infos(experiment_id=f"{k}")
])
mlflow_model_logger_dataset.save(linreg_model)
all_runs_id_end = set([
run.run_id for k in range(len(mlflow_client.list_experiments()))
for run in mlflow_client.list_run_infos(experiment_id=f"{k}")
])
assert all_runs_id_beginning == all_runs_id_end