From e95fa26a14713f181e1d6028b2202744ba73df8b Mon Sep 17 00:00:00 2001 From: Maximilian Huettenrauch Date: Wed, 27 Mar 2024 11:38:55 +0100 Subject: [PATCH] replace assert with exception in wandb logger --- tianshou/utils/logger/wandb.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tianshou/utils/logger/wandb.py b/tianshou/utils/logger/wandb.py index 213124f..a682279 100644 --- a/tianshou/utils/logger/wandb.py +++ b/tianshou/utils/logger/wandb.py @@ -83,7 +83,11 @@ class WandbLogger(BaseLogger): self.tensorboard_logger: TensorboardLogger | None = None 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) def load(self, writer: SummaryWriter) -> None: