fixing

Author: Borda

src/lightning/fabric/fabric.py

@@ -367,8 +367,7 @@ def setup_dataloaders(
             )
             for dataloader in dataloaders
         ]
-        dataloaders = dataloaders[0] if len(dataloaders) == 1 else dataloaders
-        return dataloaders  # type: ignore[return-value]
+        return dataloaders[0] if len(dataloaders) == 1 else dataloaders
 
     def _setup_dataloader(
         self, dataloader: DataLoader, use_distributed_sampler: bool = True, move_to_device: bool = True

src/lightning/pytorch/callbacks/lr_monitor.py

@@ -251,7 +251,7 @@ def _remap_keys(self, names: list[list[str]], token: str = "/pg1") -> None:
                 elif new_name not in self.lrs:
                     self.lrs[new_name] = []
 
-    def _extract_momentum(self, param_group: dict[str, list], name: str, use_betas: bool) -> dict[str, float]:
+    def _extract_momentum(self, param_group: dict[str, list[float]], name: str, use_betas: bool) -> dict[str, float]:
         if not self.log_momentum:
             return {}
 

src/lightning/pytorch/core/saving.py

@@ -184,7 +184,7 @@ def _load_state(
         obj.on_load_checkpoint(checkpoint)
 
     # load the state_dict on the model automatically
-    keys = obj.load_state_dict(checkpoint["state_dict"], strict=strict)
+    keys = obj.load_state_dict(checkpoint["state_dict"], strict=strict)  # type: ignore[arg-type]
 
     if not strict:
         if keys.missing_keys: