Zero out observations carried over from previous game

cswinter · Dec 29, 2019 · 7f46e80 · 7f46e80
1 parent 8681b64
commit 7f46e80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gym_codecraft/envs/codecraft_vec_env.py b/gym_codecraft/envs/codecraft_vec_env.py
@@ -323,8 +323,8 @@ def observe(self, env_subset=None, obs_config=None):
                 # print(f"COMPLETED {i} {game} {games[i]} == {self.games[game]} new={game_id}")
                 self.games[game] = (game_id, pid)
                 observation = codecraft.observe(game_id, pid)
-                # TODO
-                # obs[stride * i:stride * (i + 1)] = codecraft.observation_to_np(observation)
+                # TODO: use actual observation
+                obs[stride * i:stride * (i + 1)] = 0.0 # codecraft.observation_to_np(observation)
 
                 dones.append(1.0)
                 infos.append({'episode': {