update ignore

Author: zhiyuan1i

src/lightning/pytorch/strategies/ddp.py

@@ -228,7 +228,7 @@ def _register_ddp_hooks(self) -> None:
 
     def _enable_model_averaging(self) -> None:
         log.debug(f"{self.__class__.__name__}: reinitializing optimizers with post localSGD")
-        if self._model_averaging_period is None:
+        if self._model_averaging_period is None: # type: ignore[no-untyped-def]
             raise ValueError(
                 "Post-localSGD algorithm is used, but model averaging period is not provided to DDP strategy."
             )
@@ -418,7 +418,7 @@ def teardown(self) -> None:
 
         super().teardown()
 
-    def _create_stream_context(self, device_ids=None):
+    def _create_stream_context(self, device_ids=None): # type: ignore[no-untyped-def]
         """Create a stream context for the current device, if supported."""
 
         torch_lib = getattr(torch, self.root_device.type)

src/lightning/pytorch/strategies/deepspeed.py

@@ -320,7 +320,7 @@ def __init__(
         self.min_loss_scale = min_loss_scale
 
         try:
-            self.device_type = self.accelerator.get_device_type()
+            self.device_type = self.accelerator.get_device_type() # type: ignore[union-attr]
         except Exception:
             self.device_type = "cuda"
         self.torch_lib = getattr(torch, self.device_type)