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): ...@@ -66,7 +66,7 @@ class NeuralMMOEval(gym.Env):
def step(self): def step(self):
self.observations, dones, rewards, _ = self.env.step(self.actions) self.observations, dones, rewards, _ = self.env.step(self.actions)
for agent in dones: for agent in dones:
if dones[agent] is not False: if dones[agent] == -1:
self.dead_agents.append(agent) self.dead_agents.append(agent)
self.actions = self.get_agent_actions() self.actions = self.get_agent_actions()
self.alive_agents = list(self.observations.keys()) 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