Can you add data_partition to the docstring?