Skip to content

Commit d391548

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent b8bf989 commit d391548

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

tests/tests_fabric/test_connector.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@
3636
LightningEnvironment,
3737
LSFEnvironment,
3838
SLURMEnvironment,
39-
TorchElasticEnvironment, XLAEnvironment,
39+
TorchElasticEnvironment,
40+
XLAEnvironment,
4041
)
4142
from lightning.fabric.plugins.io import TorchCheckpointIO
4243
from lightning.fabric.strategies import (

tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,6 @@
2020
import pytest
2121
import torch
2222
import torch.distributed
23-
24-
from lightning.pytorch.accelerators import TPUAccelerator
2523
from lightning_utilities.core.imports import package_available
2624

2725
import lightning.pytorch
@@ -30,9 +28,11 @@
3028
LightningEnvironment,
3129
LSFEnvironment,
3230
SLURMEnvironment,
33-
TorchElasticEnvironment, XLAEnvironment,
31+
TorchElasticEnvironment,
32+
XLAEnvironment,
3433
)
3534
from lightning.pytorch import Trainer
35+
from lightning.pytorch.accelerators import TPUAccelerator
3636
from lightning.pytorch.accelerators.accelerator import Accelerator
3737
from lightning.pytorch.accelerators.cpu import CPUAccelerator
3838
from lightning.pytorch.accelerators.cuda import CUDAAccelerator
@@ -44,7 +44,9 @@
4444
DDPStrategy,
4545
DeepSpeedStrategy,
4646
FSDPStrategy,
47-
SingleDeviceStrategy, XLAStrategy, SingleTPUStrategy,
47+
SingleDeviceStrategy,
48+
SingleTPUStrategy,
49+
XLAStrategy,
4850
)
4951
from lightning.pytorch.strategies.ddp_spawn import _DDP_FORK_ALIASES
5052
from lightning.pytorch.strategies.hpu_parallel import HPUParallelStrategy

0 commit comments

Comments
 (0)