From what i remember i think we had Toggle on/off option for water java version, i remember that i was able to hide the water when i was working around the clifs or underwater.
I did ask in discord and checked all the commands in game but i just cant find anything to toggle on/off the water, if there is such thing already in game please let me know, thanks.