hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

replace assert with exception in wandb logger

Browse Source
This commit is contained in:
Maximilian Huettenrauch 2024-03-27 11:38:55 +01:00
parent dffe8cddf6
commit e95fa26a14
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tianshou/utils/logger/wandb.py
View File

@ -83,7 +83,11 @@ class WandbLogger(BaseLogger):
self.tensorboard_logger: TensorboardLogger | None = None self.tensorboard_logger: TensorboardLogger | None = None
def prepare_dict_for_logging(self, log_data: dict) -> dict[str, VALID_LOG_VALS_TYPE]: def prepare_dict_for_logging(self, log_data: dict) -> dict[str, VALID_LOG_VALS_TYPE]:
assert self.tensorboard_logger is not None if self.tensorboard_logger is None:
raise Exception(
"`logger` needs to load the Tensorboard Writer before "
"preparing data for logging. Try `logger.load(SummaryWriter(log_path))`",
)
return self.tensorboard_logger.prepare_dict_for_logging(log_data) return self.tensorboard_logger.prepare_dict_for_logging(log_data)
def load(self, writer: SummaryWriter) -> None: def load(self, writer: SummaryWriter) -> None: