fix increase_load_count (#7165)

huggingface · Sep 24, 2024 · 548d2d2 · 548d2d2
1 parent b032702
commit 548d2d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/datasets/load.py b/src/datasets/load.py
@@ -278,7 +278,7 @@ def increase_load_count(name: str):
         try:
             get_session().head(
                 "/".join((config.S3_DATASETS_BUCKET_PREFIX, name, name + ".py")),
-                user_agent=get_datasets_user_agent(),
+                headers={"User-Agent": get_datasets_user_agent()},
                 timeout=3,
             )
         except Exception: