hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified the memorymaze environment

Browse Source
This commit is contained in:
NM512
2023-08-16 21:54:09 +09:00
parent 606ec8af8c
commit 7607a92d71
3 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
envs/memorymaze.py
View File

@@ -6,12 +6,8 @@ import numpy as np
class MemoryMaze:
def __init__(self, task, obs_key="image", act_key="action", size=(64, 64), seed=0):
if task == "9x9":
self._env = gym.make("memory_maze:MemoryMaze-9x9-v0", seed=seed)
elif task == "15x15":
self._env = gym.make("memory_maze:MemoryMaze-15x15-v0", seed=seed)
else:
raise NotImplementedError(task)
# 9x9, 11x11, 13x13 and 15x15 are available
self._env = gym.make(f"memory_maze:MemoryMaze-{task}-v0", seed=seed)
self._obs_is_dict = hasattr(self._env.observation_space, "spaces")
self._obs_key = obs_key
self._act_key = act_key