move module_optimizers to types file

Author: awaelchli

src/pytorch_lightning/core/module.py

@@ -52,14 +52,13 @@
     LRSchedulerPLType,
     LRSchedulerTypeUnion,
     STEP_OUTPUT,
+    MODULE_OPTIMIZERS
 )
 from pytorch_lightning.utilities.warnings import WarningCache
 
 warning_cache = WarningCache()
 log = logging.getLogger(__name__)
 
-MODULE_OPTIMIZERS = Union[Optimizer, LightningOptimizer, List[Optimizer], List[LightningOptimizer]]
-
 
 class LightningModule(
     DeviceDtypeModuleMixin,

src/pytorch_lightning/utilities/types.py

@@ -29,6 +29,8 @@
 from torchmetrics import Metric
 from typing_extensions import Protocol, runtime_checkable
 
+from pytorch_lightning.core.optimizer import LightningOptimizer
+
 _NUMBER = Union[int, float]
 _METRIC = Union[Metric, Tensor, _NUMBER]
 _METRIC_COLLECTION = Union[_METRIC, Mapping[str, _METRIC]]
@@ -50,6 +52,7 @@
 EVAL_DATALOADERS = Union[DataLoader, Sequence[DataLoader]]
 _DEVICE = Union[torch.device, str, int]
 _MAP_LOCATION_TYPE = Optional[Union[_DEVICE, Callable[[_DEVICE], _DEVICE], Dict[_DEVICE, _DEVICE]]]
+MODULE_OPTIMIZERS = Union[Optimizer, LightningOptimizer, List[Optimizer], List[LightningOptimizer]]
 
 
 @runtime_checkable