Allow picking of partitions for producers