“California has not applied for a federal waiver created during the Trump administration that would bypass the Medicaid IMD rule limiting mental health facility size.”