diff options
Diffstat (limited to 'kafka/consumer.py')
| -rw-r--r-- | kafka/consumer.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index 7636a83..d811147 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -10,6 +10,8 @@ import numbers from threading import Lock from multiprocessing import Process, Queue as MPQueue, Event, Value +import six + try: from Queue import Empty, Queue except ImportError: # python 2 @@ -427,7 +429,7 @@ class SimpleConsumer(Consumer): for p in self.fetch_offsets.keys()) while partitions: requests = [] - for partition, buffer_size in partitions.iteritems(): + for partition, buffer_size in six.iteritems(partitions): requests.append(FetchRequest(self.topic, partition, self.fetch_offsets[partition], buffer_size)) |