update tests

pytorch · Nov 21, 2024 · f33111c · f33111c
1 parent de9dbf2
commit f33111c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/torchtune/_cli/test_download.py b/tests/torchtune/_cli/test_download.py
@@ -34,7 +34,7 @@ def snapshot_download(self, mocker, tmpdir):
 
     def test_download_calls_snapshot(self, capsys, monkeypatch, snapshot_download):
         model = "meta-llama/Llama-2-7b"
-        testargs = f"tune download {model}".split()
+        testargs = f"tune download {model} --ignore-patterns *.safetensors".split()
         monkeypatch.setattr(sys, "argv", testargs)
 
         # Call the first time and get GatedRepoError