Skip to content

Commit e92252b

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

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/lightning/pytorch/utilities/consolidate_checkpoint.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33

44
import torch
55

6-
from lightning.fabric.utilities.load import _load_distributed_checkpoint
76
from lightning.fabric.utilities.consolidate_checkpoint import _parse_cli_args, _process_cli_args
7+
from lightning.fabric.utilities.load import _load_distributed_checkpoint
88

99

1010
def _format_checkpoint(checkpoint: Dict[str, Any]) -> Dict[str, Any]:

tests/tests_pytorch/strategies/test_fsdp.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
from lightning.pytorch.plugins.precision.fsdp import FSDPPrecision
2727
from lightning.pytorch.strategies import FSDPStrategy
2828
from lightning.pytorch.trainer.states import TrainerFn
29-
from lightning.pytorch.utilities.exceptions import MisconfigurationException
3029
from lightning.pytorch.utilities.consolidate_checkpoint import _format_checkpoint
30+
from lightning.pytorch.utilities.exceptions import MisconfigurationException
3131
from torch.distributed.fsdp.fully_sharded_data_parallel import CPUOffload, FullyShardedDataParallel, MixedPrecision
3232
from torch.distributed.fsdp.wrap import always_wrap_policy, size_based_auto_wrap_policy, wrap
3333
from torchmetrics import Accuracy

0 commit comments

Comments
 (0)