diff --git a/examples/demo.py b/examples/demo.py index 8d128ea5fcb7cd6250c266e84af82a115b148c1b..9311159b14770e5a8ca2d06b9120cdd682477ad4 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -71,7 +71,7 @@ class Scenario_Generator: print("File does not exist:", filename, " Working directory: ", os.getcwd()) return env - + class Demo: def __init__(self, env): @@ -106,6 +106,8 @@ class Demo: # Action for iAgent in range(self.env.get_num_agents()): + + action = 2 if False: @@ -125,9 +127,10 @@ class Demo: trial += 1 if trial > 10: break + action_dict.update({iAgent: action}) - self.renderer.renderEnv(show=True, action_dict=action_dict) + self.renderer.renderEnv(show=True) # Environment step next_obs, all_rewards, done, _ = self.env.step(action_dict)