From d1f4d5c709b7903811a7edae7c0bca8381563002 Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmail.com>
Date: Sat, 22 Jul 2023 21:17:26 +0900
Subject: [PATCH] erased unnecessary wrapper

---
 dreamer.py | 12 ------------
 1 file changed, 12 deletions(-)

diff --git a/dreamer.py b/dreamer.py
index 0a98d3f..8c5a91a 100644
--- a/dreamer.py
+++ b/dreamer.py
@@ -228,18 +228,6 @@ def make_env(config, logger, mode, train_eps, eval_eps):
     env = wrappers.TimeLimit(env, config.time_limit)
     env = wrappers.SelectAction(env, key="action")
     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)
     return env