summaryrefslogtreecommitdiff
path: root/kafka/consumer/fetcher.py
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/consumer/fetcher.py')
-rw-r--r--kafka/consumer/fetcher.py4
1 files changed, 2 insertions, 2 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]