Type-ignore training-envs
This commit is contained in:
@@ -49,7 +49,10 @@ class TensorboardCallback(BaseCallback):
|
|||||||
local_info = self.locals["infos"][0]
|
local_info = self.locals["infos"][0]
|
||||||
if self.training_env is None:
|
if self.training_env is None:
|
||||||
return True
|
return True
|
||||||
tensorboard_metrics = self.training_env.envs[0].unwrapped.tensorboard_metrics
|
|
||||||
|
tensorboard_metrics = (
|
||||||
|
self.training_env.envs[0].unwrapped.tensorboard_metrics # type: ignore[attr-defined]
|
||||||
|
)
|
||||||
|
|
||||||
for metric in local_info:
|
for metric in local_info:
|
||||||
if metric not in ["episode", "terminal_observation"]:
|
if metric not in ["episode", "terminal_observation"]:
|
||||||
|
|||||||
Reference in New Issue
Block a user