If you are noticing your mind wondering over to what your partner is doing wrong - consider redirecting that energy back to yourself! Often it's easy to see what other people can fix - but it can be offensive and rude - if you aren't their coach!
We all strike the ball differently - we have varying confidence and skill levels and learn at different rates and in different styles. These are things that a professional teacher recognizes.
It's not your job to tell your partner how to hit the ball - only strategize about the actual game!