diff --git a/pyrit/common/download_hf_model.py b/pyrit/common/download_hf_model.py index 947dbaf8e..dbd1189a6 100644 --- 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 index 358e7f75d..b18cf7f81 100644 --- 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(""))