steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-22 03:31:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

DQN Monitor fixed

Browse Source
This commit is contained in:
steffen-illium
2021-06-02 13:36:20 +02:00
parent 017a94d6b7
commit d8e6bfc9a9
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
environments/logging/monitor.py
View File

@ -72,7 +72,8 @@ class MonitorCallback(BaseCallback):
self._monitor_dict[self.num_timesteps] = {key: val for key, val in info.items()
if key not in ['terminal_observation', 'episode']}
for env_idx, done in enumerate(self.locals.get('dones', [])):
for env_idx, done in list(enumerate(self.locals.get('dones', []))) + \
list(enumerate(self.locals.get('done', []))):
if done:
env_monitor_df = pd.DataFrame.from_dict(self._monitor_dict, orient='index')
self._monitor_dict = dict()