style

Author: rekby

ydb/_topic_reader/topic_reader_asyncio.py

@@ -470,8 +470,9 @@ def _on_start_partition_session(
     def _on_partition_session_stop(
         self, message: StreamReadMessage.StopPartitionSessionRequest
     ):
-        partition = self._partition_sessions.get(message.partition_session_id)
-        if partition is None:
+        try:
+            partition = self._partition_sessions.get(message.partition_session_id)
+        except KeyError:
             # may if receive stop partition with graceful=false after response on stop partition
             # with graceful=true and remove partition from internal dictionary
             return