pytorch/torch/testing/_internal/distributed
Tsung-Hsien Lee 84f818f359 [DTensorTestbase] Fix TestFunc typing issue (#141513)
Summary: `TestFunc` is annotated as `Callable[[object], object]` which represents a callable that takes a single argument of any type (`object`) and returns a value of any type (`object`). However, in reality, `TestFunc` could be any number of arguments, as a result, the corret typing should be `Callable[[...], object]` instead which represents a callable that takes any number of arguments (including zero) and returns a value of any type (`object`).

Test Plan: Contbuild & OSS CI

Differential Revision: D66463705

Pull Request resolved: https://github.com/pytorch/pytorch/pull/141513
Approved by: https://github.com/wz337, https://github.com/Skylion007
2024-11-26 19:48:34 +00:00
..
_shard
_tensor [DTensorTestbase] Fix TestFunc typing issue (#141513) 2024-11-26 19:48:34 +00:00
nn Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
rpc [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
__init__.py
checkpoint_utils.py
common_state_dict.py [DCP] Fixes the stateless optimizer issue of distributed state_dict (#135535) 2024-09-10 03:10:00 +00:00
ddp_under_dist_autograd_test.py Fix ROCm skip decorator for test_ddp_tp and multiprocess UTs (#136161) 2024-09-18 11:01:23 +00:00
distributed_test.py [Dist][CI] Easier override of destroy-upon-exit setting (#141192) 2024-11-21 07:32:56 +00:00
distributed_utils.py
fake_pg.py
multi_threaded_pg.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
rpc_utils.py