From 11735980745d4106444b8ba6ecb5b1a66b697b01 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Wed, 9 Aug 2017 04:46:53 +0300 Subject: [PATCH] Check annotations. --- aiogram/dispatcher/storage.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aiogram/dispatcher/storage.py b/aiogram/dispatcher/storage.py index 4deb3b54..c643c4b5 100644 --- a/aiogram/dispatcher/storage.py +++ b/aiogram/dispatcher/storage.py @@ -188,7 +188,7 @@ class FSMContext: async def update_data(self, data: typing.Dict = None, **kwargs): await self.storage.update_data(chat=self.chat, user=self.user, data=data, **kwargs) - async def set_state(self, state: typing.AnyStr): + async def set_state(self, state: typing.Union[typing.AnyStr, None]): await self.storage.set_state(chat=self.chat, user=self.user, state=state) async def set_data(self, data: typing.Dict = None): @@ -230,7 +230,7 @@ class DisabledStorage(BaseStorage): async def set_state(self, *, chat: typing.Union[str, int, None] = None, user: typing.Union[str, int, None] = None, - state: typing.AnyStr = None): + state: typing.Optional[typing.AnyStr] = None): pass async def set_data(self, *,