typing: wandb bug causes mypy error

ZettaAI · Aug 30, 2024 · d2834b2 · d2834b2
1 parent 7748b1c
commit d2834b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,