Yes you’re correct. There is nothing in Islamic teachings or practice of Ahlulbayt that says for two months one should be in this state.
It is highly advisable for the wife to be attractive to her husband and clean at all times, but not on specific days of Shahadat of Ahlulbayt (eg actual day of martyrdom only)