Fix parameter order in documentation for partition_data functions