Skip to content

Commit d5ab32e

Browse files
authored
Skip Gemma3 tests when HF_TOKEN not set (#812)
1 parent cb9de7a commit d5ab32e

1 file changed

Lines changed: 15 additions & 4 deletions

File tree

backends/candle/tests/test_gemma3.rs

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,25 @@
1-
mod common;
2-
3-
use crate::common::{sort_embeddings, SnapshotEmbeddings};
41
use anyhow::Result;
5-
use common::{batch, cosine_matcher, download_artifacts, load_tokenizer};
2+
63
use text_embeddings_backend_candle::CandleBackend;
74
use text_embeddings_backend_core::{Backend, ModelType, Pool};
85

6+
mod common;
7+
use crate::common::{
8+
batch, cosine_matcher, download_artifacts, load_tokenizer, sort_embeddings, SnapshotEmbeddings,
9+
};
10+
911
#[test]
1012
#[serial_test::serial]
1113
fn test_gemma3() -> Result<()> {
14+
// NOTE: Given that `google/embeddinggemma-300m` is a gated model, this test requires the
15+
// `HF_TOKEN` environment variable to be set
16+
if std::env::var("HF_TOKEN").is_err()
17+
|| std::env::var("HF_TOKEN").is_ok_and(|token| token.is_empty())
18+
{
19+
tracing::info!("Skipping `test_gemma3` because `HF_TOKEN` is either not set or set empty.");
20+
return Ok(());
21+
}
22+
1223
let (model_root, dense_paths) = download_artifacts("google/embeddinggemma-300m", None, None)?;
1324
let tokenizer = load_tokenizer(&model_root)?;
1425

0 commit comments

Comments
 (0)