typing

Author: Borda

src/lightning/pytorch/callbacks/lr_monitor.py

@@ -255,7 +255,8 @@ def _extract_momentum(self, param_group: dict[str, list[float]], name: str, use_
         if not self.log_momentum:
             return {}
 
-        momentum = param_group["betas"][0] if use_betas else param_group.get("momentum", 0)
+        momentum = param_group["betas"][0] if use_betas else param_group.get("momentum")
+        momentum = momentum if momentum is not None else 0
         self.last_momentum_values[name] = momentum
         return {name: momentum}
 

src/lightning/pytorch/trainer/connectors/logger_connector/result.py

@@ -92,7 +92,7 @@ def _generate_sync_fn(self) -> None:
         fn = self.no_op if self.fn is None or not self.should or self.rank_zero_only else self.fn
         # save the function as `_fn` as the meta are being re-created and the object references need to match.
         # ignore typing, bad support for `partial`: mypy/issues/1484
-        self._fn: Callable = partial(fn, reduce_op=self.op, group=self.group)  # type: ignore[arg-type,operator,misc]
+        self._fn: Callable = partial(fn, reduce_op=self.op, group=self.group)  # type: ignore[unused-ignore]
 
     @property
     def __call__(self) -> Any: