diff options
Diffstat (limited to 'kafka/consumer/group.py')
| -rw-r--r-- | kafka/consumer/group.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/kafka/consumer/group.py b/kafka/consumer/group.py index 5eeaddd..efadde1 100644 --- a/kafka/consumer/group.py +++ b/kafka/consumer/group.py @@ -842,7 +842,7 @@ class KafkaConsumer(six.Iterator): ret = self.poll(timeout_ms=self.config['consumer_timeout_ms'], max_records=1) if not ret: raise StopIteration - assert len(ret.keys()) == 1 + assert len(ret) == 1 (messages,) = ret.values() assert len(messages) == 1 return messages[0] |