try to fix types for cli

pytorch_lightning/utilities/types.py

@@ -21,6 +21,7 @@
 
 import torch
 from torch.optim import Optimizer
+from torch.optim.lr_scheduler import _LRScheduler as _TorchLRScheduler, ReduceLROnPlateau as TorchReduceLROnPlateau
 from torch.utils.data import DataLoader
 from torchmetrics import Metric
 from typing_extensions import TypedDict
@@ -48,7 +49,7 @@
 
 # Copied from `torch.optim.lr_scheduler.pyi`
 # Missing attributes were added to improve typing
-class _LRScheduler:
+class _LRScheduler(_TorchLRScheduler):
     optimizer: Optimizer
 
     def __init__(self, optimizer: Optimizer, last_epoch: int = ...) -> None:
@@ -72,7 +73,7 @@ def step(self, epoch: Optional[int] = ...) -> None:
 
 # Copied from `torch.optim.lr_scheduler.pyi`
 # Missing attributes were added to improve typing
-class ReduceLROnPlateau:
+class ReduceLROnPlateau(TorchReduceLROnPlateau):
     in_cooldown: bool
     optimizer: Optimizer