review fixes

Author: vgvoleg

ydb/_topic_reader/datatypes.py

@@ -180,22 +180,22 @@ def _pop(self) -> Tuple[List[PublicMessage], bool]:
         msgs_left = True if len(self.messages) > 1 else False
         return self.messages.pop(0), msgs_left
 
-    def _pop_batch(self, size: int) -> PublicBatch:
+    def _pop_batch(self, message_count: int) -> PublicBatch:
         initial_length = len(self.messages)
 
-        if size >= initial_length:
+        if message_count >= initial_length:
             raise ValueError("Pop batch with size >= actual size is not supported.")
 
         one_message_size = self._bytes_size // initial_length
 
         new_batch = PublicBatch(
-            messages=self.messages[:size],
+            messages=self.messages[:message_count],
             _partition_session=self._partition_session,
-            _bytes_size=one_message_size * size,
+            _bytes_size=one_message_size * message_count,
             _codec=self._codec,
         )
 
-        self.messages = self.messages[size:]
+        self.messages = self.messages[message_count:]
         self._bytes_size = self._bytes_size - new_batch._bytes_size
 
         return new_batch

ydb/_topic_reader/topic_reader_asyncio.py

@@ -401,7 +401,7 @@ def receive_batch_nowait(self, max_messages: Optional[int] = None):
             self._buffer_release_bytes(batch._bytes_size)
             return batch
 
-        cutted_batch = batch._pop_batch(size=max_messages)
+        cutted_batch = batch._pop_batch(message_count=max_messages)
 
         self._message_batches[part_sess_id] = batch
         self._buffer_release_bytes(cutted_batch._bytes_size)