env v0.12

2023-06-13 21:39:04 +08:00
parent 5038a91aad
commit b9120a7440
4 changed files with 24 additions and 8 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -215,7 +215,7 @@ def make_env(config, logger, mode, train_eps, eval_eps):
        env = gym.make('memory_maze:MemoryMaze-9x9-v0')
        from envs.memmazeEnv import MZGymWrapper
        env = MZGymWrapper(env)
-        
+        #from envs.memmazeEnv import OneHotAction as OneHotAction2
        env = wrappers.OneHotAction(env)
    elif suite == "---------mazed":
        from memory_maze import tasks