From f33111c1a423f44ed8ef779f6da14370b57489c3 Mon Sep 17 00:00:00 2001 From: Felipe Mello Date: Thu, 21 Nov 2024 09:17:00 -0800 Subject: [PATCH] update tests --- tests/torchtune/_cli/test_download.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/torchtune/_cli/test_download.py b/tests/torchtune/_cli/test_download.py index 3338adc87..f89b91a84 100644 --- 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