diff options
Diffstat (limited to 'kafka/consumer.py')
| -rw-r--r-- | kafka/consumer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index 0935dd2..51f446c 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -82,7 +82,7 @@ class Consumer(object): self.offsets = {} if not partitions: - partitions = self.client.topic_partitions[topic] + partitions = self.client.get_partition_ids_for_topic(topic) else: assert all(isinstance(x, numbers.Integral) for x in partitions) @@ -108,7 +108,7 @@ class Consumer(object): def fetch_last_known_offsets(self, partitions=None): if not partitions: - partitions = self.client.topic_partitions[self.topic] + partitions = self.client.get_partition_ids_for_topic(self.topic) def get_or_init_offset_callback(resp): try: |