From 5038a91aad6d7bc885cc73dff8df9d7ae40a0801 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=BE=B7=E7=A5=A5?=
 <zhangdexiang@zhangdexiangdeMacBook-Pro.local>
Date: Tue, 13 Jun 2023 10:44:54 +0800
Subject: [PATCH] env v0.11

---
 dreamer.py                         | 2 +-
 envs/{memmaze.py => memmazeEnv.py} | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)
 rename envs/{memmaze.py => memmazeEnv.py} (99%)

diff --git a/dreamer.py b/dreamer.py
index 3b98259..61ae538 100644
--- 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)
diff --git a/envs/memmaze.py b/envs/memmazeEnv.py
similarity index 99%
rename from envs/memmaze.py
rename to envs/memmazeEnv.py
index 9de34c4..baabf90 100644
--- a/envs/memmaze.py
+++ b/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