Baritone error's out saying "wrong Y coordinate" or something else when it goes into water, maybe you could add a water feature where it will swim, and rise up to the top of the water while still swimming to prevent drowning, also, you might want to add some calculations with flowing water if you ever add this.

This is a duplicate. Please search before creating an issue next time.