You're right that I was confused about the effect of being in a state where Amazon already has to collect sales tax. (I was thinking of tax-minimizing consumers, rather than the affiliates). When Amazon has a presence that requires them to pay sales taxes in any case, they have no credible threat to end affiliate payments. So, it's safe to remain an affiliate in those places. A happy result for Amazon, affiliates, and the States, apparently.
California and others want that same deal.
In New York, Amazon threatened to pull out but is temporarily paying sales tax (some reports suggest into escrow) and maintaining the affiliate program pending a legal challenge. Amazon may yet wind up losing their challenge, in which case they would have to decide whether to treat NY like other states where they collect sales tax – which would render being an affiliate safe – or to follow through on their affiliate freeze-out.
New York has essentially chosen the same course as California, so it will likely be resolved the same way in both states. Other states, likely including Texas and Massachusetts pending current legislative battles, are following this approach as well.
My point is: this isn't just a California-shooting-itself-in-the-foot situation, and a company can't just jump to anywhere-but-California to get back into the affiliate program. Many states, including many other top tech and business centers, are treating Amazon the same as California.