From d2834b23c4b0800ce8fdc482b0ec0aeff18c6bf0 Mon Sep 17 00:00:00 2001 From: Nico Kemnitz Date: Fri, 30 Aug 2024 10:05:21 +0200 Subject: [PATCH] typing: wandb bug causes mypy error --- zetta_utils/training/lightning/trainers/default.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zetta_utils/training/lightning/trainers/default.py b/zetta_utils/training/lightning/trainers/default.py index fe8bf62a8..026e3f739 100644 --- a/zetta_utils/training/lightning/trainers/default.py +++ b/zetta_utils/training/lightning/trainers/default.py @@ -234,7 +234,7 @@ def __init__( def on_fit_start(self, trainer: pl.Trainer, pl_module: pl.LightningModule): if not os.environ.get("WANDB_MODE", None) == "offline": # pragma: no cover api_key = os.environ.get("WANDB_API_KEY", None) - wandb.login(key=api_key) + wandb.login(key=api_key) # type: ignore[attr-defined] # Bug in wandb 0.17.8 (#8217) trainer.logger = WandbLogger( project=self.exp_name,