The answer to your question is state specific.
Certain states, my state of Ohio, prohibit a Gyn from treating the partner. According to the CDC it is permissible in Iowa. But it also said it was ok in Ohio. So I suggest talking to your Gyn.