Commit eba0ec66 authored by Siddhartha Laghuvarapu's avatar Siddhartha Laghuvarapu
Browse files

Fix env behavior for dead agents

parent cd1f812f
......@@ -66,7 +66,7 @@ class NeuralMMOEval(gym.Env):
def step(self):
self.observations, dones, rewards, _ = self.env.step(self.actions)
for agent in dones:
if dones[agent] is not False:
if dones[agent] == -1:
self.actions = self.get_agent_actions()
self.alive_agents = list(self.observations.keys())
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment