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

for more information, see https://pre-commit.ci
Lightning-AI · Dec 20, 2024 · 933036d · 933036d
1 parent 4dd9e25
commit 933036d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/lightning/fabric/strategies/ddp.py b/src/lightning/fabric/strategies/ddp.py
@@ -245,6 +245,7 @@ def _create_stream_context(self, device_ids=None):
             ctx = self.torch_lib.stream(stream)
         else:
             from contextlib import nullcontext
+
             ctx = nullcontext()
         return ctx
 

diff --git a/src/lightning/pytorch/strategies/ddp.py b/src/lightning/pytorch/strategies/ddp.py
@@ -434,6 +434,7 @@ def _create_stream_context(self, device_ids=None):
             ctx = self.torch_lib.stream(stream)
         else:
             from contextlib import nullcontext
+
             ctx = nullcontext()
         return ctx