Skip to content

Commit a468d92

Browse files
committed
fix.
Signed-off-by: Yuchuan <[email protected]>
1 parent 958f457 commit a468d92

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

tests/python/test_grammar_matcher_structural_tag.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,7 @@
88
from transformers import AutoTokenizer
99

1010
import xgrammar as xgr
11-
from xgrammar.testing import (
12-
GrammarFunctor,
13-
_get_masked_tokens_from_bitmask,
14-
_is_grammar_accept_string,
15-
)
11+
from xgrammar.testing import _get_masked_tokens_from_bitmask, _is_grammar_accept_string
1612

1713

1814
def test_utf8():

tests/python/test_structural_tag_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from transformers import AutoTokenizer
77

88
import xgrammar as xgr
9-
from xgrammar.testing import GrammarFunctor, _is_grammar_accept_string
9+
from xgrammar.testing import _is_grammar_accept_string
1010

1111
PROFILER_ON = True
1212
tokenizer_id = "meta-llama/Llama-3.1-8B-Instruct"

0 commit comments

Comments
 (0)