This does not eliminate a little flirting, though.
You can also attempt investing some time alone with your partner right after therapy. You have to keep "impressing" somebody to keep them. There is normally an excellent offer of difficult feelings and you need time to let those wounds heal.