mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
7457d139c5
Adds suppressions to pyrefly will typecheck clean: https://github.com/pytorch/pytorch/issues/163283 One more PR after this one. Test plan: dmypy restart && python3 scripts/lintrunner.py -a pyrefly check step 1: delete lines in the pyrefly.toml file from the project-excludes field step 2: run pyrefly check step 3: add suppressions, clean up unused suppressions before: https://gist.github.com/maggiemoss/4b3bf2037014e116bc00706a16aef199 after: INFO 0 errors (6,884 ignored) Pull Request resolved: https://github.com/pytorch/pytorch/pull/165002 Approved by: https://github.com/oulgen
22 lines
828 B
Python
22 lines
828 B
Python
from . import _extension
|
|
from .api import CheckpointException
|
|
from .default_planner import DefaultLoadPlanner, DefaultSavePlanner
|
|
from .filesystem import FileSystemReader, FileSystemWriter
|
|
from .hf_storage import HuggingFaceStorageReader, HuggingFaceStorageWriter
|
|
from .metadata import (
|
|
BytesStorageMetadata,
|
|
ChunkStorageMetadata,
|
|
Metadata,
|
|
TensorStorageMetadata,
|
|
)
|
|
from .optimizer import load_sharded_optimizer_state_dict
|
|
from .planner import LoadPlan, LoadPlanner, ReadItem, SavePlan, SavePlanner, WriteItem
|
|
from .quantized_hf_storage import QuantizedHuggingFaceStorageReader
|
|
|
|
# pyrefly: ignore # deprecated
|
|
from .state_dict_loader import load, load_state_dict
|
|
|
|
# pyrefly: ignore # deprecated
|
|
from .state_dict_saver import async_save, save, save_state_dict
|
|
from .storage import StorageReader, StorageWriter
|