Reinforcement learning for water system control