From 152415f32e640ce8ab99d7ea4f8dba03a7fcaadf Mon Sep 17 00:00:00 2001
From: zdx <179363811@qq.com>
Date: Sat, 17 Jun 2023 23:59:05 +0800
Subject: [PATCH] mem maze env ok 1.1

---
 configs.yaml | 2 +-
 dreamer.py   | 6 ++----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/configs.yaml b/configs.yaml
index 4db75dc..4750ee3 100644
--- a/configs.yaml
+++ b/configs.yaml
@@ -157,7 +157,7 @@ debug:
   batch_length: 20
 
 mazegym:
-  task: 9
+  task: '9'
   steps: 1e6
   action_repeat: 2
 
diff --git a/dreamer.py b/dreamer.py
index c20e8ca..3e90050 100644
--- a/dreamer.py
+++ b/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)