Distressed Democrats and relaxed Republicans? Partisanship and mental health during the COVID-19 pandemic