summaryrefslogtreecommitdiff
path: root/kafka/consumer/fetcher.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2016-09-17 18:43:38 -0700
committerDana Powers <dana.powers@gmail.com>2016-09-19 21:07:14 -0700
commitdb49c8fbdd657b652d1c3715044e6b10af6ec77b (patch)
tree9d0cd488cf9d013635c3f3791259386e7d23021b /kafka/consumer/fetcher.py
parentb05c8da2626bd639455f1ab3c25b84c59ad43690 (diff)
downloadkafka-python-max_poll_records.tar.gz
Fixup iteratorsmax_poll_records
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]