fix increase_load_count (#7165)

lhoestq · web-flow · commit 548d2d2a7ed3 · 2024-09-24T15:48:00.000+02:00
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:

Original file line number	Diff line number	Diff line change
`@@ -278,7 +278,7 @@ def increase_load_count(name: str):`
`278`	`278`	`try:`
`279`	`279`	`get_session().head(`
`280`	`280`	`"/".join((config.S3_DATASETS_BUCKET_PREFIX, name, name + ".py")),`
`281`		`- user_agent=get_datasets_user_agent(),`
	`281`	`+ headers={"User-Agent": get_datasets_user_agent()},`
`282`	`282`	`timeout=3,`
`283`	`283`	`)`
`284`	`284`	`except Exception:`