diff --git a/baselines/common/retro_wrappers.py b/baselines/common/retro_wrappers.py index 3eb2eb3..1e98044 100644 --- a/baselines/common/retro_wrappers.py +++ b/baselines/common/retro_wrappers.py @@ -132,10 +132,8 @@ class MovieRecord(gym.Wrapper): self.epcount = 0 def reset(self): if self.epcount % self.k == 0: - print('saving movie this episode', self.savedir) self.env.unwrapped.movie_path = self.savedir else: - print('not saving this episode') self.env.unwrapped.movie_path = None self.env.unwrapped.movie = None self.epcount += 1