diff --git a/examples/demo.py b/examples/demo.py index 25e9b8cbf52c45d4afa9cc76b5456ec211fb8f73..01ef58a2c27da22f4567f2b4722310c85c49e0ba 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -107,7 +107,6 @@ class Demo: # environment step (apply the actions to all agents) next_obs, all_rewards, done, _ = self.env.step(action_dict) - if done['__all__']: break