Revert "Fixed test_regex to expect the final state"

This reverts commit 37d9e01.
dottxt-ai · Feb 5, 2024 · 11f2286 · 11f2286
1 parent f6e6743
commit 11f2286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/fsm/test_fsm.py b/tests/fsm/test_fsm.py
@@ -46,7 +46,7 @@ def convert_token_to_string(self, token):
     tokenizer = MockTokenizer()
     fsm = RegexFSM(regex_str, tokenizer)
 
-    assert fsm.states_to_token_maps == {0: {1: 1}, 1: {3: 1}}
+    assert fsm.states_to_token_maps == {0: {1: 1}}
     assert fsm.allowed_token_ids(state=0) == [1]
     assert fsm.next_state(state=0, token_id=1) == 1
     assert fsm.next_state(state=0, token_id=tokenizer.eos_token_id) == fsm.final_state