fix: rename _assert_exactly_one to _validate_init and make it not static

AI21Labs · Jan 2, 2024 · 5384fd4 · 5384fd4
1 parent 5bb5938
commit 5384fd4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/ai21_tokenizer/jurassic_tokenizer.py b/ai21_tokenizer/jurassic_tokenizer.py
@@ -23,7 +23,7 @@ def __init__(
         model_file_handle: Optional[BinaryIO] = None,
         config: Optional[Dict[str, Any]] = None,
     ):
-        JurassicTokenizer._assert_exactly_one(model_path, model_file_handle)
+        self._validate_init(model_path, model_file_handle)
 
         model_proto = load_binary(model_path) if model_path else model_file_handle.read()
 
@@ -67,8 +67,7 @@ def from_file_handle(
     def from_file_path(cls, model_path: PathLike, config: Optional[Dict[str, Any]] = None) -> JurassicTokenizer:
         return cls(model_path=model_path, config=config)
 
-    @staticmethod
-    def _assert_exactly_one(model_path: Optional[PathLike], model_file_handle: Optional[BinaryIO]) -> None:
+    def _validate_init(self, model_path: Optional[PathLike], model_file_handle: Optional[BinaryIO]) -> None:
         if model_path is None and model_file_handle is None:
             raise ValueError("Must provide exactly one of model_path or model_file_handle. Got none.")