Added missing import

noamgat · Sep 28, 2024 · 48bc632 · 48bc632
1 parent 0b4f308
commit 48bc632
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -2,12 +2,12 @@
 from pstats import Stats
 from typing import Optional
 from transformers import AutoTokenizer, PreTrainedTokenizerBase
-
 from lmformatenforcer import CharacterLevelParser
 from lmformatenforcer.exceptions import LMFormatEnforcerException
 from lmformatenforcer.tokenenforcer import TokenEnforcer, TokenEnforcerTokenizerData
 from lmformatenforcer.integrations.transformers import build_token_enforcer_tokenizer_data
-
+import logging
+
 
 _tokenizer: Optional[PreTrainedTokenizerBase] = None
 _tokenizer_data: Optional[TokenEnforcerTokenizerData] = None