From 3adaf11555b42a64026ed0bfae0b89cebca3e210 Mon Sep 17 00:00:00 2001 From: wenfeiy-db Date: Tue, 14 Nov 2023 13:19:20 -0800 Subject: [PATCH] clean --- tests/test_hf_conversion_script.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/tests/test_hf_conversion_script.py b/tests/test_hf_conversion_script.py index 73a027704c..d2c2a9e1c9 100644 --- a/tests/test_hf_conversion_script.py +++ b/tests/test_hf_conversion_script.py @@ -425,18 +425,6 @@ def test_huggingface_conversion_callback(model: str, tmp_path: pathlib.Path, else 0) assert mlflow_logger_mock.register_model.call_count == ( 1 if log_to_mlflow else 0) - # if log_to_mlflow: - # # mlflow_logger_mock.save_model.assert_called_with( - # # flavor='transformers', - # # transformers_model=ANY, - # # path=ANY, - # # task='text-generation', - # # metatdata={'task': 'llm/v1/completions'} - # # ) - # assert mlflow_logger_mock.register_model.call_count == 1 - # else: - # assert mlflow_logger_mock.save_model.call_count == 0 - # assert mlflow_logger_mock.register_model.call_count == 0 else: assert mlflow_logger_mock.log_model.call_count == 0 assert mlflow_logger_mock.register_model.call_count == 0