fix mypy issue

Author: deependujha

src/lightning/fabric/plugins/precision/deepspeed.py

@@ -19,7 +19,7 @@
 from torch.nn import Module
 from typing_extensions import override
 
-from lightning.fabric.plugins.precision.precision import Precision
+from lightning.fabric.plugins.precision.precision import _PRECISION_INPUT_STR, Precision
 from lightning.fabric.utilities.imports import _raise_enterprise_not_available
 from lightning.fabric.utilities.types import Steppable
 
@@ -84,11 +84,11 @@ def optimizer_step(
         return self.deepspeed_impl.optimizer_step(optimizer, **kwargs)
 
     @property
-    def precision(self) -> _PRECISION_INPUT:
+    def precision(self) -> _PRECISION_INPUT_STR:
         return self.deepspeed_impl.precision
 
     @precision.setter
-    def precision(self, precision: _PRECISION_INPUT) -> None:
+    def precision(self, precision: _PRECISION_INPUT_STR) -> None:
         self.deepspeed_impl.precision = precision
 
     @property

src/lightning/fabric/plugins/precision/xla.py

@@ -16,7 +16,7 @@
 import torch
 from typing_extensions import override
 
-from lightning.fabric.plugins.precision.precision import Precision
+from lightning.fabric.plugins.precision.precision import _PRECISION_INPUT_STR, Precision
 from lightning.fabric.utilities.imports import _raise_enterprise_not_available
 from lightning.fabric.utilities.types import Optimizable
 
@@ -63,9 +63,9 @@ def _desired_dtype(self, dtype: torch.dtype) -> None:
         self.xla_impl._desired_dtype = dtype
 
     @property
-    def precision(self) -> _PRECISION_INPUT:
+    def precision(self) -> _PRECISION_INPUT_STR:
         return self.xla_impl.precision
 
     @precision.setter
-    def precision(self, precision: _PRECISION_INPUT) -> None:
+    def precision(self, precision: _PRECISION_INPUT_STR) -> None:
         self.xla_impl.precision = precision

src/lightning/pytorch/loggers/neptune.py

@@ -271,7 +271,9 @@ def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None:
 
     @override
     @rank_zero_only
-    def log_metrics(self, metrics: dict[str, Union[Tensor, float]], step: Optional[int] = None) -> None:
+    def log_metrics(  # type: ignore[override]
+        self, metrics: dict[str, Union[Tensor, float]], step: Optional[int] = None
+    ) -> None:
         """Log metrics (numeric values) in Neptune runs.
 
         Args:

src/lightning/pytorch/plugins/precision/deepspeed.py

@@ -21,7 +21,7 @@
 from typing_extensions import override
 
 import lightning.pytorch as pl
-from lightning.fabric.plugins.precision.deepspeed import _PRECISION_INPUT
+from lightning.fabric.plugins.precision.deepspeed import _PRECISION_INPUT, _PRECISION_INPUT_STR
 from lightning.fabric.utilities.imports import _raise_enterprise_not_available
 from lightning.fabric.utilities.types import Steppable
 from lightning.pytorch.plugins.precision.precision import Precision
@@ -111,11 +111,11 @@ def clip_gradients(
         )
 
     @property
-    def precision(self) -> str:
+    def precision(self) -> _PRECISION_INPUT_STR:
         return self.deepspeed_precision_impl.precision
 
     @precision.setter
-    def precision(self, value: str) -> None:
+    def precision(self, value: _PRECISION_INPUT_STR) -> None:
         self.deepspeed_precision_impl.precision = value
 
     @property

src/lightning/pytorch/plugins/precision/xla.py

@@ -17,7 +17,7 @@
 from typing_extensions import override
 
 import lightning.pytorch as pl
-from lightning.fabric.plugins.precision.xla import _PRECISION_INPUT
+from lightning.fabric.plugins.precision.xla import _PRECISION_INPUT, _PRECISION_INPUT_STR
 from lightning.fabric.utilities.imports import _raise_enterprise_not_available
 from lightning.fabric.utilities.types import Optimizable
 from lightning.pytorch.plugins.precision.precision import Precision
@@ -54,11 +54,11 @@ def optimizer_step(  # type: ignore[override]
         return self.xla_impl.optimizer_step(optimizer, model, closure, **kwargs)
 
     @property
-    def precision(self) -> _PRECISION_INPUT:
+    def precision(self) -> _PRECISION_INPUT_STR:
         return self.xla_impl.precision
 
     @precision.setter
-    def precision(self, precision: _PRECISION_INPUT) -> None:
+    def precision(self, precision: _PRECISION_INPUT_STR) -> None:
         self.xla_impl.precision = precision
 
     @property