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/envs/memmazeEnv.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