Which one is better for 7 months old baby to wear for swimming pool? Charlie banana swim diaper or Swim suit for baby?

A thermal swim wear and a swim diaper for young babies. Baby catches cold quite easily especially if the pool is not a heated pool. Thus, a thermal swim wear is a good investment.