clean

mosaicml · Nov 14, 2023 · 3adaf11 · 3adaf11
1 parent 5e0d31b
commit 3adaf11
Showing 1 changed file with 0 additions and 12 deletions.
diff --git 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