For the water ones, I have to ask if it's okay to immerse? All surgery I have had said not to for weeks at a time. If you ask them about bathing they say to plastic wrap but swimming is several steps too far; not allowed to fully immerse no matter how well protected.
Regardless of which you go with, Go Slow. Zero weight means even getting into the pool/machine. We get used to using our whole bodies, so get a spotter, don't be shy about using the crazy pool lift, etc. No toughing it out. Follow the instructions to the letter to keep from re-injuring it. Have had too many friends who fucked themselves up forever for not letting shit heal right.