Skip to content

Commit 7ec8a6b

Browse files
pre-commit-ci[bot]zhixiangli
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7c7a732 commit 7ec8a6b

2 files changed

Lines changed: 12 additions & 12 deletions

File tree

src/lightning/fabric/strategies/fsdp.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,17 @@
4848
_Sharded,
4949
_validate_keys_for_strict_loading,
5050
)
51+
from lightning.fabric.utilities.cloud_io import (
52+
_atomic_save,
53+
_checkpoint_join,
54+
_is_checkpoint_dir,
55+
_is_local_file_protocol,
56+
_load,
57+
_prepare_directory_checkpoint,
58+
_remove_checkpoint,
59+
_resolve_path,
60+
get_filesystem,
61+
)
5162
from lightning.fabric.utilities.distributed import (
5263
ReduceOp,
5364
_distributed_is_initialized,
@@ -60,17 +71,6 @@
6071
_TORCH_GREATER_EQUAL_2_2,
6172
_TORCH_GREATER_EQUAL_2_3,
6273
)
63-
from lightning.fabric.utilities.cloud_io import (
64-
_atomic_save,
65-
_checkpoint_join,
66-
_is_checkpoint_dir,
67-
_is_local_file_protocol,
68-
_load,
69-
_prepare_directory_checkpoint,
70-
_remove_checkpoint,
71-
_resolve_path,
72-
get_filesystem,
73-
)
7474
from lightning.fabric.utilities.init import _has_meta_device_parameters_or_buffers
7575
from lightning.fabric.utilities.load import _METADATA_FILENAME, _lazy_load, _materialize_tensors, _move_state_into
7676
from lightning.fabric.utilities.rank_zero import rank_zero_deprecation, rank_zero_only, rank_zero_warn

src/lightning/fabric/strategies/model_parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
_BackwardSyncControl,
4444
_validate_keys_for_strict_loading,
4545
)
46+
from lightning.fabric.utilities.cloud_io import _is_local_file_protocol, _load
4647
from lightning.fabric.utilities.distributed import (
4748
ReduceOp,
4849
_distributed_is_initialized,
@@ -51,7 +52,6 @@
5152
_sync_ddp_if_available,
5253
)
5354
from lightning.fabric.utilities.distributed import group as _group
54-
from lightning.fabric.utilities.cloud_io import _is_local_file_protocol, _load
5555
from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_3, _TORCH_GREATER_EQUAL_2_4
5656
from lightning.fabric.utilities.init import _materialize_distributed_module
5757
from lightning.fabric.utilities.load import _METADATA_FILENAME, _lazy_load, _move_state_into

0 commit comments

Comments
 (0)