Merge branch 'main' of github.com:eilab-gt/NovGrid into main

eilab-gt · Jun 14, 2022 · 0d73bfe · 0d73bfe
2 parents 128805f + ffa7c18
commit 0d73bfe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
+logs
 novgrid/baselines/logs/*
 novgrid/baselines/logs
 *.pyc

diff --git a/novgrid/envs/lavagapdoorkey.py b/novgrid/envs/lavagapdoorkey.py
@@ -193,7 +193,7 @@ def step(self, action, **kwargs):
         obs, reward, done, info = super().step(action)
         if done and fwd_cell and fwd_cell.type == 'lava':
             self.agent_pos = fwd_pos
-            obs = self.gen_obs()['image']
+            obs = self.gen_obs()
             done = False
         return obs, reward, done, info