Skip to content

Commit 0798ef7

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

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/lightning/fabric/wrappers.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ def _register_backward_hook(self, tensor: Tensor) -> Tensor:
217217
hook = partial(_backward_hook, (strategy_requires or precision_requires))
218218
tensor.register_hook(hook)
219219
return tensor
220-
220+
221221
def wrap_forward_method(self, method: Callable) -> Callable:
222222
@wraps(method)
223223
def wrapper(*args: Any, **kwargs: Any) -> Any:
@@ -231,6 +231,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
231231

232232
apply_to_collection(output, dtype=Tensor, function=self._register_backward_hook)
233233
return output
234+
234235
return wrapper
235236

236237
@override

0 commit comments

Comments
 (0)