Fix tests

outlines-dev · May 10, 2024 · 32cc20d · 32cc20d
1 parent faea4cd
commit 32cc20d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/tests/fsm/test_fsm.py b/tests/fsm/test_fsm.py
@@ -82,9 +82,7 @@ def convert_token_to_string(self, token):
  assert fsm.is_final_state(state)
 
  state = fsm.next_state(state=5, token_id=103)
- assert state == 5
-
- assert fsm.is_final_state(-1)
+ assert fsm.is_final_state(state)
 
 
 def test_cfg():

diff --git a/tests/fsm/test_guide.py b/tests/fsm/test_guide.py
@@ -180,9 +180,7 @@ def convert_token_to_string(self, token):
  assert fsm.is_final_state(state)
 
  state = fsm.get_next_state(state=5, token_id=103)
- assert state == 5
-
- assert fsm.is_final_state(-1)
+ assert fsm.is_final_state(state)
 
 
 def test_cfg():