Skip to content

Commit 05023f6

Browse files
fix
Signed-off-by: Nathan Hoos <thwackyy.y@gmail.com>
1 parent 53be045 commit 05023f6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/model_executor/test_guided_processors.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ async def test_guided_logits_processor_with_reasoning(
153153
# ever get to the Guide (which will error if invalid tokenid's are input)
154154
# because the ReasoningParser is causing the logits processor
155155
# to quick return.
156-
json_lp(token_ids, tensor)
156+
regex_lp(token_ids, tensor)
157157
assert tensor.shape == original_tensor.shape
158158
assert torch.allclose(tensor, original_tensor)
159159

@@ -169,7 +169,7 @@ async def test_guided_logits_processor_with_reasoning(
169169
assert json_lp is not None
170170
tensor = torch.rand(32000)
171171
original_tensor = torch.clone(tensor)
172-
regex_lp(token_ids, tensor)
172+
json_lp(token_ids, tensor)
173173
assert tensor.shape == original_tensor.shape
174174
assert torch.allclose(tensor, original_tensor)
175175

0 commit comments

Comments
 (0)