Skip to content
This repository was archived by the owner on Nov 13, 2024. It is now read-only.

Commit 181762c

Browse files
authored
Support optional imports in unit tests (#293)
1 parent c9ce812 commit 181762c

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

tests/unit/tokenizer/test_cohere_hf_tokenizer.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,14 @@ class TestCohereHFTokenizer(BaseTestTokenizer):
88
@staticmethod
99
@pytest.fixture(scope="class")
1010
def tokenizer():
11-
return CohereHFTokenizer()
11+
try:
12+
tokenizer = CohereHFTokenizer()
13+
except ImportError:
14+
pytest.skip(
15+
"`cohere` extra not installed. Skipping CohereHFTokenizer unit "
16+
"tests"
17+
)
18+
return tokenizer
1219

1320
@staticmethod
1421
@pytest.fixture

tests/unit/tokenizer/test_llama_tokenizer.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,14 @@ class TestLlamaTokenizer(BaseTestTokenizer):
88
@staticmethod
99
@pytest.fixture(scope="class")
1010
def tokenizer():
11-
return LlamaTokenizer(model_name="hf-internal-testing/llama-tokenizer")
11+
try:
12+
tokenizer = LlamaTokenizer(model_name="hf-internal-testing/llama-tokenizer")
13+
except ImportError:
14+
pytest.skip(
15+
"`transformers` extra not installed. Skipping LLamaTokenizer unit "
16+
"tests"
17+
)
18+
return tokenizer
1219

1320
@staticmethod
1421
@pytest.fixture

0 commit comments

Comments
 (0)