Handle HF_TOKEN in ApiBuilder for candle/tests (#724)

alvarobartt · web-flow · commit d22a2c2a9f09 · 2025-09-19T21:04:09.000+02:00
diff --git a/backends/candle/tests/common.rs b/backends/candle/tests/common.rs
@@ -133,12 +133,12 @@ pub fn download_artifacts(
 ) -> Result<(PathBuf, Option<Vec<String>>)> {
     let mut builder = ApiBuilder::from_env().with_progress(false);
 
-    if let Some(cache_dir) = std::env::var_os("HUGGINGFACE_HUB_CACHE") {
-        builder = builder.with_cache_dir(cache_dir.into());
+    if let Ok(token) = std::env::var("HF_TOKEN") {
+        builder = builder.with_token(Some(token));
     }
 
-    if let Ok(origin) = std::env::var("HF_HUB_USER_AGENT_ORIGIN") {
-        builder = builder.with_user_agent("origin", origin.as_str());
+    if let Some(cache_dir) = std::env::var_os("HUGGINGFACE_HUB_CACHE") {
+        builder = builder.with_cache_dir(cache_dir.into());
     }
 
     let api = builder.build().unwrap();

Original file line number	Diff line number	Diff line change
`@@ -133,12 +133,12 @@ pub fn download_artifacts(`
`133`	`133`	`) -> Result<(PathBuf, Option<Vec<String>>)> {`
`134`	`134`	`let mut builder = ApiBuilder::from_env().with_progress(false);`
`135`	`135`
`136`		`- if let Some(cache_dir) = std::env::var_os("HUGGINGFACE_HUB_CACHE") {`
`137`		`- builder = builder.with_cache_dir(cache_dir.into());`
	`136`	`+ if let Ok(token) = std::env::var("HF_TOKEN") {`
	`137`	`+ builder = builder.with_token(Some(token));`
`138`	`138`	`}`
`139`	`139`
`140`		`- if let Ok(origin) = std::env::var("HF_HUB_USER_AGENT_ORIGIN") {`
`141`		`- builder = builder.with_user_agent("origin", origin.as_str());`
	`140`	`+ if let Some(cache_dir) = std::env::var_os("HUGGINGFACE_HUB_CACHE") {`
	`141`	`+ builder = builder.with_cache_dir(cache_dir.into());`
`142`	`142`	`}`
`143`	`143`
`144`	`144`	`let api = builder.build().unwrap();`