diff options
Diffstat (limited to 'kafka')
| -rw-r--r-- | kafka/consumer/fetcher.py | 4 | ||||
| -rw-r--r-- | kafka/consumer/group.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/kafka/consumer/fetcher.py b/kafka/consumer/fetcher.py index 62190fb..15fa1c9 100644 --- a/kafka/consumer/fetcher.py +++ b/kafka/consumer/fetcher.py @@ -448,10 +448,10 @@ class Fetcher(six.Iterator): return self def __next__(self): - ret = self.fetched_records(max_records=1) + ret, _ = self.fetched_records(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] 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] |