Programmatically set partition in python IDE

When debugging python externally using my IDE of choice. how do I set the default partition for my interpreter session?

e.g.

from dataiku.base import remoterun
remoterun.set_dku_env_var_and_sys_env_var('DKU_CURRENT_PROJECT_KEY', 'MY_PROJECT')

allows me to set the default project for my session. I'm looking for the partition equivalent. as if I were running the recipe for a particular scenario.

I don't wnat to have to do Dataset.add_read_partitions() for every dataset (or pepper those lines into my recipe code just so I can run the recipe from my IDE)