modified the memorymaze environment

2023-08-16 21:54:09 +09:00
parent 606ec8af8c
commit 7607a92d71
3 changed files with 10 additions and 18 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -216,7 +216,7 @@ def make_env(config, mode):
            seed=config.seed,
        )
        env = wrappers.OneHotAction(env)
-    elif suite == "MemoryMaze":
+    elif suite == "memorymaze":
        from envs.memorymaze import MemoryMaze

        env = MemoryMaze(task, seed=config.seed)