Skip to content

Commit 2f8f2b5

Browse files
ralphbeanclaude
andcommitted
[Tests] Remove cache_dir parameter from test fixture
The cache_dir parameter was removed from ModelArguments in favor of environment variables. This commit updates the test fixture to no longer pass cache_dir to AutoTokenizer.from_pretrained(). Fixes the AttributeError in test_dataset_loading.py and test_registry.py tests that were failing in CI/CD. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]> Signed-off-by: Ralph Bean <[email protected]>
1 parent 2376e9a commit 2f8f2b5

File tree

1 file changed

+0
-1
lines changed
  • tests/llmcompressor/transformers/finetune/data

1 file changed

+0
-1
lines changed

tests/llmcompressor/transformers/finetune/data/conftest.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ def tiny_llama_model_args(tiny_llama_path):
1818
def tiny_llama_tokenizer(tiny_llama_model_args):
1919
tokenizer = AutoTokenizer.from_pretrained(
2020
tiny_llama_model_args.model,
21-
cache_dir=tiny_llama_model_args.cache_dir,
2221
use_fast=True,
2322
revision=tiny_llama_model_args.model_revision,
2423
use_auth_token=True if tiny_llama_model_args.use_auth_token else None,

0 commit comments

Comments
 (0)