remove return value from download_specific_files

Azure · Nov 6, 2024 · 43bf1c3 · 43bf1c3
1 parent 2097639
commit 43bf1c3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/pyrit/common/download_hf_model.py b/pyrit/common/download_hf_model.py
@@ -58,8 +58,6 @@ def download_specific_files(model_id: str, file_patterns: list, token: str, cach
     # Download the files
     download_files(urls, token, cache_dir)
 
-    return urls
-
 
 def download_files(urls: list, token: str, cache_dir: Path):
     headers = {"Authorization": f"Bearer {token}"}

diff --git a/tests/test_hf_model_downloads.py b/tests/test_hf_model_downloads.py
@@ -38,5 +38,4 @@ def test_download_specific_files(setup_environment):
 
     with patch("os.makedirs"):
         with patch("pyrit.common.download_hf_model.download_files"):
-            urls = download_specific_files(MODEL_ID, FILE_PATTERNS, token, Path(""))
-            assert urls
+            download_specific_files(MODEL_ID, FILE_PATTERNS, token, Path(""))