diff --git a/tests/conftest.py b/tests/conftest.py index 8ab8fd6a4..964d73bd6 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -58,11 +58,15 @@ def snapshot_download_w_retry(*args, **kwargs): """ with hf_offline_context(True): try: - return snapshot_download(*args, **kwargs) + return snapshot_download( + *args, user_agent={"is_ci": "true", "axolotl": "ci"}, **kwargs + ) except LocalEntryNotFoundError: pass with hf_offline_context(False): - return snapshot_download(*args, **kwargs) + return snapshot_download( + *args, user_agent={"is_ci": "true", "axolotl": "ci"}, **kwargs + ) @pytest.fixture(scope="session", autouse=True)