From b40b80519203ffb7ecdd638d8e0af52019d4689d Mon Sep 17 00:00:00 2001 From: Jerry Chen Date: Sun, 14 Jan 2024 11:17:14 -0800 Subject: [PATCH] fix mock (#872) --- tests/a_scripts/inference/test_convert_composer_to_hf.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py index 28fb9219f8..deed181475 100644 --- a/tests/a_scripts/inference/test_convert_composer_to_hf.py +++ b/tests/a_scripts/inference/test_convert_composer_to_hf.py @@ -300,6 +300,8 @@ def test_huggingface_conversion_callback_interval( mlflow_logger_mock.save_model = MagicMock() mlflow_logger_mock.register_model = MagicMock() mlflow_logger_mock.model_registry_prefix = '' + mlflow_logger_mock._experiment_id = 'mlflow-experiment-id' + mlflow_logger_mock._run_id = 'mlflow-run-id' trainer = Trainer( model=original_model, device='gpu', @@ -534,6 +536,8 @@ def test_huggingface_conversion_callback( mlflow_logger_mock.save_model = MagicMock() mlflow_logger_mock.register_model = MagicMock() mlflow_logger_mock.model_registry_prefix = '' + mlflow_logger_mock._experiment_id = 'mlflow-experiment-id' + mlflow_logger_mock._run_id = 'mlflow-run-id' trainer = Trainer( model=original_model, device='gpu',