fix(stub): fix stub

Author: ChieloNewctle

general_sam/general_sam.pyi

@@ -1,11 +1,13 @@
 from typing import Callable, Mapping, Optional, Sequence, Union
 
+ByteOrChar = Union[str, int]
+
 class TrieNode:
     def is_in_chars(self) -> bool: ...
     def is_in_bytes(self) -> bool: ...
     def get_node_id(self) -> int: ...
     def is_accepting(self) -> bool: ...
-    def get_trans(self) -> Mapping[Union[str, int], int]: ...
+    def get_trans(self) -> Mapping[ByteOrChar, int]: ...
     def get_parent(self) -> int: ...
 
 class Trie:
@@ -23,13 +25,13 @@ class Trie:
     def get_node(self, node_id: int) -> Optional[TrieNode]: ...
     def dfs_travel(
         self,
-        in_stack_callback: Callable[[int, Optional[str]], None],
+        in_stack_callback: Callable[[int, Optional[ByteOrChar]], None],
         out_stack_callback: Callable[[int], None],
         root_node_id: Optional[int] = None,
     ) -> TrieNode: ...
     def bfs_travel(
         self,
-        in_queue_callback: Callable[[int, Optional[str]], None],
+        in_queue_callback: Callable[[int, Optional[ByteOrChar]], None],
         out_queue_callback: Callable[[int], None],
         root_node_id: Optional[int] = None,
     ) -> TrieNode: ...
@@ -41,7 +43,7 @@ class GeneralSAMState:
     def is_nil(self) -> bool: ...
     def is_root(self) -> bool: ...
     def is_accepting(self) -> bool: ...
-    def get_trans(self) -> Mapping[Union[str, int], int]: ...
+    def get_trans(self) -> Mapping[ByteOrChar, int]: ...
     def get_suffix_parent_id(self) -> int: ...
     def copy(self) -> 'GeneralSAMState': ...
     def goto_suffix_parent(self): ...
@@ -52,14 +54,18 @@ class GeneralSAMState:
     def dfs_along(
         self,
         trie: Trie,
-        in_stack_callback: Callable[['GeneralSAMState', int, Optional[str]], None],
+        in_stack_callback: Callable[
+            ['GeneralSAMState', int, Optional[ByteOrChar]], None
+        ],
         out_stack_callback: Callable[['GeneralSAMState', int], None],
         trie_node_id: Optional[int] = None,
     ) -> TrieNode: ...
     def bfs_along(
         self,
         trie: Trie,
-        in_queue_callback: Callable[['GeneralSAMState', int, Optional[str]], None],
+        in_queue_callback: Callable[
+            ['GeneralSAMState', int, Optional[ByteOrChar]], None
+        ],
         out_queue_callback: Callable[['GeneralSAMState', int], None],
         trie_node_id: Optional[int] = None,
     ) -> TrieNode: ...