Skip to content

Commit b1d75d0

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

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/lightning/pytorch/core/module.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -675,15 +675,17 @@ def all_gather(
675675
all_gather = self.trainer.strategy.all_gather
676676
data = convert_to_tensors(data, device=self.device)
677677
return apply_to_collection(data, Tensor, all_gather, group=group, sync_grads=sync_grads)
678-
678+
679679
@override
680680
def forward(self, *inputs) -> Any:
681681
r"""Same as :meth:`torch.nn.Module.forward`.
682+
682683
Args:
683684
*args: Whatever you decide to pass into the forward method.
684685
**kwargs: Keyword arguments are also possible.
685686
Return:
686687
Your model's output
688+
687689
"""
688690
return super().forward(*inputs)
689691

0 commit comments

Comments
 (0)