If you indicate the resident state for a partner, won't ProSeries give you a notification that you might need to prepare the K-1 for that resident state? Then you would add that state. I use Lacerte.

There must be a way to avoid manual calculations.  If you are truly facing that, then perhaps a Fixed Asset software would be useful.