diff options
Diffstat (limited to 'kafka/partitioner.py')
| -rw-r--r-- | kafka/partitioner.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/kafka/partitioner.py b/kafka/partitioner.py index 84db4d5..817e416 100644 --- a/kafka/partitioner.py +++ b/kafka/partitioner.py @@ -31,7 +31,8 @@ class RoundRobinPartitioner(Partitioner): in a round robin fashion """ def __init__(self, partitions): - self._set_partitions(partitions) + super(RoundRobinPartitioner, self).__init__(partitions) + self.iterpart = cycle(partitions) def _set_partitions(self, partitions): self.partitions = partitions |