The RP2040 PIO has a feature called "side-set" that can set the levels or dirs of pins as part of any PIO instruction. However, it appears that only the levels are supported, and there's no way to use side-set to set pin directions. Here's the relevant quote from the datasheet:

It seems like this should be a configuration parameter to rp2.StateMachine().