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

applied formatter

Browse Source
This commit is contained in:
NM512
2023-06-18 16:57:05 +09:00
parent 775eb94e7f
commit e3329b35e5
2 changed files with 74 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dreamer.py
View File

@@ -213,13 +213,15 @@ def make_env(config, logger, mode, train_eps, eval_eps):
env = wrappers.OneHotAction(env)
elif suite == "MemoryMaze":
import gym
if task == '9x9':
env = gym.make('memory_maze:MemoryMaze-9x9-v0')
elif task == '15x15':
env = gym.make('memory_maze:MemoryMaze-15x15-v0')
if task == "9x9":
env = gym.make("memory_maze:MemoryMaze-9x9-v0")
elif task == "15x15":
env = gym.make("memory_maze:MemoryMaze-15x15-v0")
else:
raise NotImplementedError(suite)
from envs.memorymaze import MemoryMaze
env = MemoryMaze(env)
env = wrappers.OneHotAction(env)
elif suite == "crafter":