Fix cases of single partition

mahendra · mahendra · commit c13ee1df6ab2 · 2013-06-28T13:59:06.000+05:30
diff --git a/README.md b/README.md
@@ -59,6 +59,8 @@ producer = KeyedProducer(kafka, "my-topic", partitioner=RoundRobinPartitioner)
 
 ## Multiprocess consumer
 ```python
+from kafka.consume import MultiProcessConsumer
+
 # This will split the number of partitions among two processes
 consumer = MultiProcessConsumer(kafka, "my-topic", "my-group", num_procs=2)
 
diff --git a/kafka/consumer.py b/kafka/consumer.py
@@ -437,11 +437,12 @@ def __init__(self, client, group, topic, auto_commit=True,
                 partitions_per_proc += 1
 
         # The final set of chunks
-        chunks = map(None, *[iter(partitions)] * int(partitions_per_proc))
+        chunker = lambda *x: [] + list(x)
+        chunks = map(chunker, *[iter(partitions)] * int(partitions_per_proc))
 
         self.procs = []
         for chunk in chunks:
-            chunk = filter(lambda x: x is not None, list(chunk))
+            chunk = filter(lambda x: x is not None, chunk)
             proc = Process(target=self._consume, args=(chunk,))
             proc.daemon = True
             proc.start()
