fix fix fix

mosaicml · Jan 10, 2024 · 62015a7 · 62015a7
1 parent 433b616
commit 62015a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/callbacks/test_hf_checkpointer.py b/tests/callbacks/test_hf_checkpointer.py
@@ -45,7 +45,8 @@ def assert_checkpoint_saves_to_uri(
     with patch('logging.Logger.info', dummy_log_info(logs)):
         my_checkpointer = HuggingFaceCheckpointer(
             save_folder=uri, save_interval=dummy_save_interval)
-        my_checkpointer.remote_ud.upload_file = dummy_upload_file
+        if my_checkpointer.remote_ud is not None:
+            my_checkpointer.remote_ud.upload_file = dummy_upload_file
         my_checkpointer._save_checkpoint(dummy_state, dummy_logger)
 
     assert any([uri_base in str(log) for log in logs])