Skip to content

Commit a84b9b1

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

File tree

1 file changed

+2
-5
lines changed
  • src/lightning/fabric/utilities

1 file changed

+2
-5
lines changed

src/lightning/fabric/utilities/init.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -116,11 +116,8 @@ def _has_meta_device_parameters_or_buffers(obj: Union[Module, Optimizer], recurs
116116

117117

118118
def _has_all_dtensor_params_or_buffers(obj: Union[Module, Optimizer], recurse: bool = True) -> bool:
119-
"""
120-
Check whether all parameters and buffers of a given
121-
:class:`torch.nn.Module` or :class:`torch.optim.Optimizer` are instances of
122-
:class:`torch.distributed.tensor.DTensor`.
123-
"""
119+
"""Check whether all parameters and buffers of a given :class:`torch.nn.Module` or :class:`torch.optim.Optimizer`
120+
are instances of :class:`torch.distributed.tensor.DTensor`."""
124121
from torch.distributed.tensor import DTensor
125122

126123
if isinstance(obj, Optimizer):

0 commit comments

Comments
 (0)