From 0c1009bfc49b759fe75eed4fd377b4fbad534ad5 Mon Sep 17 00:00:00 2001 From: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com> Date: Fri, 19 Sep 2025 14:17:28 +0200 Subject: [PATCH] Handle `HF_TOKEN` in `ApiBuilder` for `candle/tests` --- backends/candle/tests/common.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backends/candle/tests/common.rs b/backends/candle/tests/common.rs index 43d7aece..896e0f65 100644 --- a/backends/candle/tests/common.rs +++ b/backends/candle/tests/common.rs @@ -133,12 +133,12 @@ pub fn download_artifacts( ) -> Result<(PathBuf, Option>)> { 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();