Update README.md

README.md

@@ -50,9 +50,9 @@ print(f"mean_reward={mean_reward:.2f} +/- {std_reward}")
 
 # Enjoy trained agent
 vec_env = model.get_env()
-obs = vec_env.reset()
+obs, info = vec_env.reset()
 for _ in range(1000):
     action, _states = model.predict(obs, deterministic=True)
-    obs, rewards, dones, info = vec_env.step(action)
+    obs, rewards, terminated,truncated, info = vec_env.step(action)
     vec_env.render("human")
 ```