From 6a739384ee9db0c3a2cdb369351d6e89dd10d16d Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Tue, 26 Sep 2023 10:48:13 +0200 Subject: [PATCH] WandbLogger: Use less restrictive type annotation for config --- tianshou/utils/logger/wandb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tianshou/utils/logger/wandb.py b/tianshou/utils/logger/wandb.py index bf49c48..e984d46 100644 --- a/tianshou/utils/logger/wandb.py +++ b/tianshou/utils/logger/wandb.py @@ -52,7 +52,7 @@ class WandbLogger(BaseLogger): name: str | None = None, entity: str | None = None, run_id: str | None = None, - config: argparse.Namespace | None = None, + config: argparse.Namespace | dict | None = None, monitor_gym: bool = True, ) -> None: super().__init__(train_interval, test_interval, update_interval)