env v0.11

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)

envs/memmazeEnv.py

@@ -9,7 +9,7 @@ import gym
 import numpy as np
 
 
-class GymWrapper:
+class MZGymWrapper:
 
   def __init__(self, env, obs_key='image', act_key='action'):
     self._env = env