diff options
Diffstat (limited to 'kafka/consumer.py')
-rw-r--r-- | kafka/consumer.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/kafka/consumer.py b/kafka/consumer.py index 3f8d8c2..98f18a0 100644 --- a/kafka/consumer.py +++ b/kafka/consumer.py @@ -3,6 +3,7 @@ from __future__ import absolute_import from itertools import izip_longest, repeat import logging import time +import numbers from threading import Lock from multiprocessing import Process, Queue as MPQueue, Event, Value from Queue import Empty, Queue @@ -81,6 +82,8 @@ class Consumer(object): if not partitions: partitions = self.client.topic_partitions[topic] + else: + assert all(isinstance(x, numbers.Integral) for x in partitions) # Variables for handling offset commits self.commit_lock = Lock() |