Fixed test

predibase · Feb 21, 2024 · a81ce89 · a81ce89
1 parent f8dc86a
commit a81ce89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/tests/models/test_causal_lm.py b/server/tests/models/test_causal_lm.py
@@ -145,7 +145,7 @@ def test_causal_lm_batch_type(default_causal_lm):
 
 @pytest.mark.parametrize("causal_lm_batch, generated_token_id", [
     ("default_causal_lm_batch", 13),
-    ("schema_constrained_causal_lm_batch", 198),
+    ("schema_constrained_causal_lm_batch", 90),
 ])
 def test_causal_lm_generate_token(default_causal_lm, causal_lm_batch, generated_token_id, request):
     causal_lm_batch = request.getfixturevalue(causal_lm_batch)