To set the partition to use for any dataset, you just call the server method with the partition you want to use.