File tree 1 file changed +2
-2
lines changed
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -153,7 +153,7 @@ async def test_guided_logits_processor_with_reasoning(
153
153
# ever get to the Guide (which will error if invalid tokenid's are input)
154
154
# because the ReasoningParser is causing the logits processor
155
155
# to quick return.
156
- json_lp (token_ids , tensor )
156
+ regex_lp (token_ids , tensor )
157
157
assert tensor .shape == original_tensor .shape
158
158
assert torch .allclose (tensor , original_tensor )
159
159
@@ -169,7 +169,7 @@ async def test_guided_logits_processor_with_reasoning(
169
169
assert json_lp is not None
170
170
tensor = torch .rand (32000 )
171
171
original_tensor = torch .clone (tensor )
172
- regex_lp (token_ids , tensor )
172
+ json_lp (token_ids , tensor )
173
173
assert tensor .shape == original_tensor .shape
174
174
assert torch .allclose (tensor , original_tensor )
175
175
You can’t perform that action at this time.
0 commit comments