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

erased unnecessary wrapper

Browse Source
This commit is contained in:
NM512 2023-07-22 21:17:26 +09:00
parent 03d91cb2c1
commit d1f4d5c709
1 changed files with 0 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dreamer.py
View File

@ -228,18 +228,6 @@ def make_env(config, logger, mode, train_eps, eval_eps):
env = wrappers.TimeLimit(env, config.time_limit) env = wrappers.TimeLimit(env, config.time_limit)
env = wrappers.SelectAction(env, key="action") env = wrappers.SelectAction(env, key="action")
env = wrappers.UUID(env) env = wrappers.UUID(env)
# if (mode == "train") or (mode == "eval"):
# callbacks = [
# functools.partial(
# ProcessEpisodeWrap.process_episode,
# config,
# logger,
# mode,
# train_eps,
# eval_eps,
# )
# ]
# env = wrappers.CollectDataset(env, mode, train_eps, callbacks=callbacks)
env = wrappers.RewardObs(env) env = wrappers.RewardObs(env)
return env return env