Skip to content

Commit 933036d

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

File tree

2 files changed

+2
-0
lines changed
  • src/lightning

2 files changed

+2
-0
lines changed

src/lightning/fabric/strategies/ddp.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,7 @@ def _create_stream_context(self, device_ids=None):
245245
ctx = self.torch_lib.stream(stream)
246246
else:
247247
from contextlib import nullcontext
248+
248249
ctx = nullcontext()
249250
return ctx
250251

src/lightning/pytorch/strategies/ddp.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -434,6 +434,7 @@ def _create_stream_context(self, device_ids=None):
434434
ctx = self.torch_lib.stream(stream)
435435
else:
436436
from contextlib import nullcontext
437+
437438
ctx = nullcontext()
438439
return ctx
439440

0 commit comments

Comments
 (0)