fix tests

mosaicml · Dec 4, 2024 · 6e2bdf2 · 6e2bdf2
1 parent b4947c9
commit 6e2bdf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py
@@ -553,7 +553,7 @@ def test_huggingface_conversion_callback_interval(
         mlflow_logger_mock.log_model.assert_called_with(
             transformers_model=ANY,
             flavor='transformers',
-            artifact_path=f'huggingface/{save_interval}',
+            artifact_path=f'huggingface/ba{trainer.state.timestamp.batch.value}',
             registered_model_name='dummy-registered-name',
             run_id='mlflow-run-id',
             await_registration_for=3600,