Skip to content

Commit a476e09

Browse files
brendanlongclaude
andcommitted
Fix test_cuda using nonexistent mlm_tokens fixture
The test_cuda function referenced a fixture named mlm_tokens which was never defined, causing a fixture-not-found error. Changed to use the existing tokens fixture which provides the same MLM-style tokenized input. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent f2347e3 commit a476e09

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/acceptance/test_hooked_encoder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,6 @@ def test_input_list_of_strings_mlm(our_bert, huggingface_bert, tokenizer):
222222

223223

224224
@pytest.mark.skipif(not torch.cuda.is_available(), reason="Requires a CUDA device")
225-
def test_cuda(mlm_tokens):
225+
def test_cuda(tokens):
226226
model = HookedEncoder.from_pretrained(MODEL_NAME)
227-
model(mlm_tokens)
227+
model(tokens)

0 commit comments

Comments
 (0)