I've never heard of a rehab telling a spouse to stay distant from the other spouse. Obviously, there needs to be change, and sometimes that's best achieved with some separation or downtime... in order to get established with new boundaries and new habits while working a program.
Actually, they tell people in relationships to not make any big decisions regarding the relationship during the first year... if that's possible. There is so much change going on in that first year that it's best not to make any drastic changes.
But things can't be the same they were, either. Try not to worry too much about what happened. It's very difficult being in recovery AND trying to manage a relationship. Very hard. Use the separation to keep working on yourself. Time will help make things clearer.