frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Unify TypeAlias definitions in optimizer.py (#161493)

Browse Source 
Fixes #160834

This issue unifies TypeAlias definitions in [optimizer.py](https://github.com/pytorch/pytorch/blob/main/torch/optim/optimizer.py)

This ensures the following:

- Consistency and Standardization
- Enhanced IDE support
- Prevents runtime confusion

Pull Request resolved: https://github.com/pytorch/pytorch/pull/161493
Approved by: https://github.com/Skylion007
This commit is contained in:
Parshant Sharma
2025-08-30 00:34:58 +00:00
committed by PyTorch MergeBot
parent 0f81e7f640
commit 4ad9fbc83a
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/optim/optimizer.py
View File

@ -28,9 +28,10 @@ _P = ParamSpec("_P")
Args: TypeAlias = tuple[Any, ...]
Kwargs: TypeAlias = dict[str, Any]
StateDict: TypeAlias = dict[str, Any]
DeviceDict = dict[Optional[torch.device], torch.Tensor]
DeviceDtypeDict = dict[Optional[tuple[torch.device, torch.dtype]], torch.Tensor]
DeviceDict: TypeAlias = dict[Optional[torch.device], torch.Tensor]
DeviceDtypeDict: TypeAlias = dict[
Optional[tuple[torch.device, torch.dtype]], torch.Tensor
]
GlobalOptimizerPreHook: TypeAlias = Callable[
["Optimizer", Args, Kwargs], Optional[tuple[Args, Kwargs]]