Walk speed should be determined by the number of tiles of water present on the block the player is standing on, the more the slower.
In this way the player wouldn't be slowed down that much when walking in little puddles.

I have added this feature in the upgrade to 1.17 and 1.18 I have created. Check it out: https://github.com/Felicis/Water-Physics-Overhaul