pytorch
7e0c26d4 - [JIT] Allow `tuple` and `list` generics (#98703)

Commit
1 year ago
[JIT] Allow `tuple` and `list` generics (#98703) As in Python-3.9+ `Dict`, `List`, and `Tuple` from `typing` module are deprecated in favor of their `builtins` counterparts, see [PEP 585](https://peps.python.org/pep-0585/) Test plan: Run: ``` import torch from typing import Union @torch.jit.script def to_tuple(v: Union[int, tuple[int, int]]) -> tuple[int, int]: """Converts int or tuple to tuple of ints.""" if torch.jit.isinstance(v, int): return v, v else: return v print(to_tuple(1), to_tuple((3, 4))) ``` It's almost impossible to add test to an existing CI, as test script will not be parseable by Python-3.8, which is a oldest supported Python version Fixes https://github.com/pytorch/pytorch/issues/98521 Pull Request resolved: https://github.com/pytorch/pytorch/pull/98703 Approved by: https://github.com/kit1980
Author
Committer
Parents
Loading