Fixed mypy failure on the CI (#3279)

pytorch · Sep 3, 2024 · b2f18b0 · b2f18b0
1 parent 25f8143
commit b2f18b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ignite/handlers/wandb_logger.py b/ignite/handlers/wandb_logger.py
@@ -135,13 +135,13 @@ def __init__(self, *args: Any, **kwargs: Any):
             )
         if kwargs.get("init", True):
             kwargs.pop("init", None)
-            wandb.init(*args, **kwargs)
+            self._wandb.init(*args, **kwargs)
 
     def __getattr__(self, attr: Any) -> Any:
         return getattr(self._wandb, attr)
 
     def close(self) -> None:
-        self._wandb.finish()
+        self._wandb.finish()  # type: ignore[attr-defined]
 
     def _create_output_handler(self, *args: Any, **kwargs: Any) -> "OutputHandler":
         return OutputHandler(*args, **kwargs)