I think Apple doesn't really have a choice. They've been very strongly encouraged by the current US government to move as much chip manufacturing to the US as possible, and particularly to make Intel Foundry work, or face... problems.

Also the AI boom means NVIDIA et al. can afford to buy TSMC's best processes at scale, which means less available capacity for Apple.

I'm sure given no other forces at work, Apple would prefer to stick with what they were doing previously, buying the lion's share of TSMC's best process.