K-1 Distribution - which state do i pay taxes in?

I receive distributions from a partnership throughout the year. It is a combination of return of capital as well as gains / losses. 

 

I moved in May 2021 from CA to VT. How should i divide up the partnership gains / losses by state? Is it based on the portion of the year when i moved, or the timing of when distributions happened and where I lived at each time?