saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-30 20:18:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add default filename for monitor

Browse source
This commit is contained in:
Antonin Raffin 2019-10-29 18:42:34 +01:00
parent c0cb9fc9c5
commit 9e8f6e0020
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/common/monitor.py
View file

@ -13,7 +13,7 @@ class Monitor(Wrapper):
EXT = "monitor.csv"
file_handler = None
def __init__(self, env, filename, allow_early_resets=True, reset_keywords=(), info_keywords=()):
def __init__(self, env, filename=None, allow_early_resets=True, reset_keywords=(), info_keywords=()):
"""
A monitor wrapper for Gym environments, it is used to know the episode reward, length, time and other data.