diff --git a/tools.py b/tools.py index b36ed40..9572fe1 100644 --- a/tools.py +++ b/tools.py @@ -164,7 +164,7 @@ def simulate( # replace obs with done by initial state obs[index] = result # step agents - obs = {k: np.stack([o[k] for o in obs]) for k in obs[0]} + obs = {k: np.stack([o[k] for o in obs]) for k in obs[0] if "log_" not in k} action, agent_state = agent(obs, done, agent_state) if isinstance(action, dict): action = [