improve type hint

z44d · Sep 20, 2024 · eb53ba0 · eb53ba0
1 parent 279e9be
commit eb53ba0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tgram/tgbot.py b/tgram/tgbot.py
@@ -14,7 +14,7 @@
 from .errors import APIException, MutedError
 from .utils import API_URL, get_file_name, ALL_UPDATES
 from .sync import wrap
-from .storage import KvsqliteStorage, RedisStorage
+from .storage import KvsqliteStorage, RedisStorage, StorageBase
 from .storage.utils import check_update
 from .types.type_ import Type_
 from concurrent.futures.thread import ThreadPoolExecutor
@@ -269,7 +269,7 @@ def __init__(
         self.retry_after = retry_after
         self.plugins = Path(plugins) if isinstance(plugins, str) else plugins
         self.skip_updates = skip_updates
-        self.storage: Optional[KvsqliteStorage] = None
+        self.storage: Optional[StorageBase] = None
         self.storage_client = storage_engine
 
         self.executor = ThreadPoolExecutor(self.workers, thread_name_prefix="Handlers")