mem maze env ok 1.1

configs.yaml

@@ -157,7 +157,7 @@ debug:
   batch_length: 20
 
 mazegym:
-  task: 9
+  task: '9'
   steps: 1e6
   action_repeat: 2
 

dreamer.py

@@ -212,17 +212,15 @@ def make_env(config, logger, mode, train_eps, eval_eps):
         env = wrappers.OneHotAction(env)
     elif suite == "mazegym":
         import gym
-        if task == 9:
+        if task == '9':
             env = gym.make('memory_maze:MemoryMaze-9x9-v0')
-        elif task == 15:
+        elif task == '15':
             env = gym.make('memory_maze:MemoryMaze-15x15-v0')
         else:
             raise NotImplementedError(suite)
         from envs.memmazeEnv import MZGymWrapper
         env = MZGymWrapper(env)
-        #from envs.memmazeEnv import OneHotAction as OneHotAction2
         env = wrappers.OneHotAction2(env)
-
     else:
         raise NotImplementedError(suite)
     env = wrappers.TimeLimit(env, config.time_limit)