Before revision: Surprise-based learning allows agents to adapt quickly in non-stationary stochastic environments .
After revision: Surprise-based learning allows agents to rapidly adapt to non-stationary stochastic environments .