env v0.11

2023-06-13 10:44:54 +08:00
parent 7879c6cfe7
commit 5038a91aad
2 changed files with 2 additions and 2 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -213,7 +213,7 @@ def make_env(config, logger, mode, train_eps, eval_eps):
    elif suite == "mazegym":
        import gym
        env = gym.make('memory_maze:MemoryMaze-9x9-v0')
-        from envs.memmaze import MZGymWrapper
+        from envs.memmazeEnv import MZGymWrapper
        env = MZGymWrapper(env)
        
        env = wrappers.OneHotAction(env)