reformat

mistralai · Sep 22, 2024 · 0ee589a · 0ee589a
1 parent 8c66d7f
commit 0ee589a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/mistral_common/tokens/tokenizers/base.py b/src/mistral_common/tokens/tokenizers/base.py
@@ -150,7 +150,8 @@ def __call__(self, content: Union[ImageChunk, ImageURLChunk]) -> ImageEncoding:
         ...
 
     @property
-    def image_token(self) -> int: ...
+    def image_token(self) -> int:
+        ...
 
 
 class InstructTokenizer(Generic[InstructRequestType, FIMRequestType, TokenizedType, AssistantMessageType]):
@@ -181,7 +182,8 @@ def encode_user_message(
         is_first: bool,
         system_prompt: Optional[str] = None,
         force_img_first: bool = False,
-    ) -> Tuple[List[int], List[np.ndarray]]: ...
+    ) -> Tuple[List[int], List[np.ndarray]]:
+        ...
 
     @abstractmethod
     def encode_user_content(
@@ -190,4 +192,5 @@ def encode_user_content(
         is_last: bool,
         system_prompt: Optional[str] = None,
         force_img_first: bool = False,
-    ) -> Tuple[List[int], List[np.ndarray]]: ...
+    ) -> Tuple[List[int], List[np.ndarray]]:
+        ...