Is there another swim school nearby with a mother and baby class?
At the swim school we use they don't go in the water alone until they're comfortable in the water and ok with having an instructor.