Skip to content

Commit b792073

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

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/tests_pytorch/trainer/test_ddp_sigterm_handling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,10 @@
1515

1616
pytestmark = pytest.mark.skipif(
1717
not torch.distributed.is_available() or torch.cuda.device_count() < 2,
18-
reason="Requires torch.distributed and at least 2 CUDA devices"
18+
reason="Requires torch.distributed and at least 2 CUDA devices",
1919
)
2020

21+
2122
class DummyModel(LightningModule):
2223
def training_step(self, batch, batch_idx):
2324
# Simulate SIGTERM in rank 0 at batch 2

0 commit comments

Comments
 (0)