 Watch the agent play obs = env.reset() for i in range(1000):     action, _state = model.predict(obs)     obs, reward, done, info = env.step(action)     env.render()     if done:       obs = env.reset() ``` 