fix tests

mosaicml · Jan 23, 2024 · da27b1f · da27b1f
1 parent 17e261f
commit da27b1f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/llmfoundry/utils/model_download_utils.py b/llmfoundry/utils/model_download_utils.py
@@ -102,7 +102,7 @@ def download_from_hf_hub(
             f'No supported model weights found in repo {model}.' +
             ' Please make sure the repo contains either safetensors or pytorch weights.'
         )
-
+    
     allow_patterns = TOKENIZER_FILES if tokenizers_only else None
 
     download_start = time.time()

diff --git a/tests/utils/test_model_download_utils.py b/tests/utils/test_model_download_utils.py
@@ -110,6 +110,7 @@ def test_download_from_hf_hub_weights_pref(mock_list_repo_files: MagicMock,
     mock_snapshot_download.assert_called_once_with(
         test_repo_id,
         local_dir=save_dir,
+        allow_patterns=None,
         ignore_patterns=expected_ignore_patterns,
         token=None)