Swimming for 3MO

Hi mummies, would like to bring my 3mo baby for a swim at the baby pool in my condo. Could I get advice on what to prepare for baby? Afraid it might be cold for him. We have a swim diaper. Is it sufficient? ? thanks!