Fix type annotation for AIOKafkaAdminClient.create_partitions (#978)

* ✏️ fix typo : create_partitions() method accept NewPartitions instead of TopicPartition

* update CHANGES.rst

* Update aiokafka/admin/client.py

Co-authored-by: Denis Otkidach <[email protected]>

* Optimize imports

---------

Co-authored-by: Denis Otkidach <[email protected]>

Author: alm0ra

CHANGES.rst

@@ -8,6 +8,7 @@ Unreleased
 Bugfixes:
 
 * Fix serialization for batch (issue #886, pr #887 by @ydjin0602)
+* Fix type annotation for `AIOKafkaAdminClient.create_partitions` (pr #978 by @alm0ra)
 
 
 0.10.0 (2023-12-15)

aiokafka/admin/client.py

@@ -29,6 +29,7 @@
 from aiokafka.structs import OffsetAndMetadata, TopicPartition
 
 from .config_resource import ConfigResource, ConfigResourceType
+from .new_partitions import NewPartitions
 from .new_topic import NewTopic
 from .records_to_delete import RecordsToDelete
 
@@ -395,15 +396,15 @@ def _convert_config_resources(
         return broker_resources, topic_resources
 
     @staticmethod
-    def _convert_topic_partitions(topic_partitions: Dict[str, TopicPartition]):
+    def _convert_topic_partitions(topic_partitions: Dict[str, NewPartitions]):
         return [
             (topic_name, (new_part.total_count, new_part.new_assignments))
             for topic_name, new_part in topic_partitions.items()
         ]
 
     async def create_partitions(
         self,
-        topic_partitions: Dict[str, TopicPartition],
+        topic_partitions: Dict[str, NewPartitions],
         timeout_ms: Optional[int] = None,
         validate_only: bool = False,
     ) -> Response: