new plotting, omit_agent_obs

2021-06-02 18:12:56 +02:00
parent 8810955e86
commit b72013407e
5 changed files with 79 additions and 20 deletions
--- a/environments/logging/monitor.py
+++ b/environments/logging/monitor.py
@ -12,12 +12,11 @@ class MonitorCallback(BaseCallback):

    ext = 'png'

-    def __init__(self, env, filepath=Path('debug_out/monitor.pick'), plotting=True):
+    def __init__(self, filepath=Path('debug_out/monitor.pick'), plotting=True):
        super(MonitorCallback, self).__init__()
        self.filepath = Path(filepath)
        self._monitor_df = pd.DataFrame()
        self._monitor_dict = dict()
-        self.env = env
        self.plotting = plotting
        self.started = False
        self.closed = False