There has still been a major decline in federal prosecutions for marijuana in the US, something like -50% since 2020. The vast majority of weed prosecutions happen at the state level.
State vs federal unfortunatly doesn't change the math behind the "if we do W then union X will not support us and then voting block Y will vote against us at a predicted rate of Z%" math upon which most of these political decisions are calculated.