Yuanyuan Chen a60d9e1f6d Fix flake8 B028 warnings (#166224) ... This PR fixes flake8 B028 warning by specifying stacklevel=2 in `warnings.warn`. The advantage is that users can know more contextual information about PyTorch warnings. Pull Request resolved: https://github.com/pytorch/pytorch/pull/166224 Approved by: https://github.com/ezyang		2025-10-26 06:18:55 +00:00
..
_experimental	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
examples	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
__init__.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
_async_executor.py
_async_process_executor.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
_async_thread_executor.py
_checkpointer.py	[distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256)	2025-10-17 20:14:35 +00:00
_consolidate_hf_safetensors.py	Add pg argument to consolidate_safetensors_files_on_every_rank (#161421)	2025-08-29 13:31:11 +00:00
_dedup_save_plans.py	[distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256)	2025-10-17 20:14:35 +00:00
_dedup_tensors.py
_extension.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
_fsspec_filesystem.py	[distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256)	2025-10-17 20:14:35 +00:00
_hf_utils.py
_nested_dict.py
_pg_transport.py	Fix pyrefly ignores 1/n (#166239)	2025-10-26 00:44:10 +00:00
_sharded_tensor_utils.py
_state_dict_stager.py	state dict staging fixes (#166025)	2025-10-22 23:32:41 +00:00
_storage_utils.py
_traverse.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
_version.py
api.py
default_planner.py	[distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256)	2025-10-17 20:14:35 +00:00
filesystem.py	Fix flake8 B028 warnings (#166224)	2025-10-26 06:18:55 +00:00
format_utils.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
hf_storage.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
logger.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
logging_handlers.py
metadata.py
optimizer.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
planner_helpers.py	[3/N] Import Callable from collections.abc in torch/distributed (#164104)	2025-09-30 00:28:53 +00:00
planner.py
quantized_hf_storage.py	[distributed] Replace assert statements in distributed checkpoint with explicit checks (#165256)	2025-10-17 20:14:35 +00:00
resharding.py
staging.py	Use correct pyrefly syntax in suppressions distributed/... (#166241)	2025-10-26 04:16:41 +00:00
state_dict_loader.py	Fix flake8 B028 warnings (#166224)	2025-10-26 06:18:55 +00:00
state_dict_saver.py	Fix flake8 B028 warnings (#166224)	2025-10-26 06:18:55 +00:00
state_dict.py	Fix flake8 B028 warnings (#166224)	2025-10-26 06:18:55 +00:00
stateful.py
storage.py
utils.py	Fix flake8 B028 warnings (#166224)	2025-10-26 06:18:55 +00:00