[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

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

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
-from typing import Any, Optional
+from typing import Any
 
 from typing_extensions import override
 

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

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from contextlib import AbstractContextManager
-from typing import Any, Literal, Optional
+from typing import Any, Literal
 
 import torch
 from lightning_utilities.core.imports import RequirementCache

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

@@ -13,7 +13,7 @@
 # limitations under the License.
 from collections.abc import Mapping
 from contextlib import AbstractContextManager
-from typing import TYPE_CHECKING, Any, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal, Union
 
 import torch
 from typing_extensions import override

src/lightning/fabric/strategies/deepspeed.py

@@ -12,9 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
+from collections.abc import Callable
 from contextlib import AbstractContextManager
 from datetime import timedelta
-from typing import TYPE_CHECKING, Any, Callable, Optional, Union
+from typing import TYPE_CHECKING, Any, Optional
 
 import torch
 from torch.nn import Module
@@ -331,7 +332,7 @@ def setup_optimizer(self, optimizer: Optimizer) -> Optimizer:
         return self.deepspeed_impl.setup_optimizer(optimizer=optimizer)
 
     @override
-    def module_init_context(self, empty_init: Optional[bool] = None) -> AbstractContextManager:
+    def module_init_context(self, empty_init: bool | None = None) -> AbstractContextManager:
         return self.deepspeed_impl.module_init_context(empty_init=empty_init)
 
     @override
@@ -415,9 +416,7 @@ def clip_gradients_norm(
         )
 
     @override
-    def clip_gradients_value(
-        self, module: "DeepSpeedEngine", optimizer: Optimizer, clip_val: Union[float, int]
-    ) -> None:
+    def clip_gradients_value(self, module: "DeepSpeedEngine", optimizer: Optimizer, clip_val: float | int) -> None:
         return self.deepspeed_impl.clip_gradients_value(module=module, optimizer=optimizer, clip_val=clip_val)
 
     @classmethod

src/lightning/fabric/strategies/launchers/xla.py

@@ -11,7 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import TYPE_CHECKING, Any, Callable, Union
+from collections.abc import Callable
+from typing import TYPE_CHECKING, Any, Union
 
 from typing_extensions import override
 

src/lightning/fabric/strategies/xla.py

@@ -11,7 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import TYPE_CHECKING, Any, Callable, Optional, Union
+from collections.abc import Callable
+from typing import TYPE_CHECKING, Any
 
 import torch
 from torch import Tensor
@@ -160,7 +161,7 @@ def all_reduce(
         return self.xla_strategy_impl.all_reduce(output=output, group=group, reduce_op=reduce_op)
 
     @override
-    def barrier(self, name: Optional[str] = None, *args: Any, **kwargs: Any) -> None:
+    def barrier(self, name: str | None = None, *args: Any, **kwargs: Any) -> None:
         return self.xla_strategy_impl.barrier(name=name, *args, **kwargs)
 
     @override

src/lightning/fabric/strategies/xla_fsdp.py

@@ -11,8 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from collections.abc import Callable
 from contextlib import AbstractContextManager
-from typing import TYPE_CHECKING, Any, Callable, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal, Optional
 
 import torch
 from torch import Tensor
@@ -188,7 +189,7 @@ def setup_module(self, module: Module) -> Module:
     def module_to_device(self, module: Module) -> None:
         return self.xla_fsdp_impl.module_to_device(module=module)
 
-    def module_init_context(self, empty_init: Optional[bool] = None) -> AbstractContextManager:
+    def module_init_context(self, empty_init: bool | None = None) -> AbstractContextManager:
         return self.xla_fsdp_impl.module_init_context(empty_init=empty_init)
 
     @override
@@ -251,7 +252,7 @@ def all_reduce(
         return self.xla_fsdp_impl.all_reduce(output=output, group=group, reduce_op=reduce_op)
 
     @override
-    def barrier(self, name: Optional[str] = None, *args: Any, **kwargs: Any) -> None:
+    def barrier(self, name: str | None = None, *args: Any, **kwargs: Any) -> None:
         return self.xla_fsdp_impl.barrier(name=name, *args, **kwargs)
 
     @override

src/lightning/pytorch/loggers/comet.py

@@ -237,12 +237,12 @@ def experiment(self) -> comet_experiment:
 
     @override
     @rank_zero_only
-    def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None:
+    def log_hyperparams(self, params: dict[str, Any] | Namespace) -> None:
         return self.logger_impl.log_hyperparams(params)
 
     @override
     @rank_zero_only
-    def log_metrics(self, metrics: Mapping[str, Union[Tensor, float]], step: Optional[int] = None) -> None:
+    def log_metrics(self, metrics: Mapping[str, Tensor | float], step: Optional[int] = None) -> None:
         return self.logger_impl.log_metrics(metrics, step)
 
     @override

src/lightning/pytorch/loggers/mlflow.py

@@ -19,7 +19,7 @@
 import os
 from argparse import Namespace
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, Literal, Optional, Union
+from typing import TYPE_CHECKING, Any, Literal
 
 from typing_extensions import override
 
@@ -164,12 +164,12 @@ def experiment_id(self) -> str | None:
 
     @override
     @rank_zero_only
-    def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None:
+    def log_hyperparams(self, params: dict[str, Any] | Namespace) -> None:
         return self.logger_impl.log_hyperparams(params)
 
     @override
     @rank_zero_only
-    def log_metrics(self, metrics: Mapping[str, float], step: Optional[int] = None) -> None:
+    def log_metrics(self, metrics: Mapping[str, float], step: int | None = None) -> None:
         return self.logger_impl.log_metrics(metrics, step)
 
     @override

src/lightning/pytorch/loggers/neptune.py

@@ -18,7 +18,7 @@
 
 import logging
 from argparse import Namespace
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any, Union
 
 from torch import Tensor
 from typing_extensions import override
@@ -266,13 +266,13 @@ def run(self) -> "Run":
 
     @override
     @rank_zero_only
-    def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None:
+    def log_hyperparams(self, params: dict[str, Any] | Namespace) -> None:
         return self.logger_impl.log_hyperparams(params)
 
     @override
     @rank_zero_only
     def log_metrics(  # type: ignore[override]
-        self, metrics: dict[str, Union[Tensor, float]], step: Optional[int] = None
+        self, metrics: dict[str, Tensor | float], step: int | None = None
     ) -> None:
         """Log metrics (numeric values) in Neptune runs.