def on_rebalance(cns, old_partition_offsets, new_partition_offsets):
self.assertTrue(len(new_partition_offsets) > 0)
self.assigned_called = True
for id_ in iterkeys(new_partition_offsets):
new_partition_offsets[id_] = self.offset_reset
return new_partition_offsets
def on_rebalance(cns, old_partition_offsets, new_partition_offsets):
self.assertTrue(len(new_partition_offsets) > 0)
self.assigned_called = True
for id_ in iterkeys(new_partition_offsets):
new_partition_offsets[id_] = self.offset_reset
return new_partition_offsets