summaryrefslogtreecommitdiff
path: root/kafka/consumer.py
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/consumer.py')
-rw-r--r--kafka/consumer.py3
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()