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

Enable local_partial_types (#118467)

Browse Source 
When using dmypy, this setting is enabled and cannot be turned off. Force it for regular mypy too.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118467
Approved by: https://github.com/Skylion007
ghstack dependencies: #118414, #118418, #118432
This commit is contained in:
Edward Z. Yang
2024-01-27 19:30:07 -08:00
committed by PyTorch MergeBot
parent 2ed0af2bde
commit 46712b019d
54 changed files with 163 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributed/rendezvous.py
View File

@ -9,14 +9,14 @@ import numbers
import os
import sys
from datetime import timedelta
from typing import Dict, Optional
from typing import Dict, Optional, Callable, Iterator, Tuple
from torch.distributed import FileStore, PrefixStore, Store, TCPStore
from .constants import default_pg_timeout
_rendezvous_handlers = {}
_rendezvous_handlers: Dict[str, Callable[..., Iterator[Tuple[Store, int, int]]]] = {}
def register_rendezvous_handler(scheme, handler):