From f35480f2a6821d417fa15ae94cff70d0937ef531 Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmail.com>
Date: Sun, 1 Oct 2023 06:25:23 +0900
Subject: [PATCH] policy is not given logs

---
 tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = [