User-agent on CI snapshot download

2025-05-20 08:52:33 -07:00
1 changed files with 6 additions and 2 deletions
--- 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)