Handle multiproc calls for now

This commit is contained in:
Matthias
2023-10-15 11:52:18 +02:00
parent 2d9d8dc976
commit 1e1b8dbe53

View File

@@ -50,9 +50,12 @@ class TensorboardCallback(BaseCallback):
if self.training_env is None:
return True
tensorboard_metrics = (
self.training_env.envs[0].unwrapped.tensorboard_metrics # type: ignore[attr-defined]
)
if hasattr(self.training_env, 'envs'):
tensorboard_metrics = self.training_env.envs[0].unwrapped.tensorboard_metrics
else:
# For RL-multiproc - usage of [0] might need to be evaluated
tensorboard_metrics = self.training_env.get_attr("tensorboard_metrics")[0]
for metric in local_info:
if metric not in ["episode", "terminal_observation"]: