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

PEP585 update - test (#145176)

Browse Source 
See #145101 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145176
Approved by: https://github.com/bobrenjc93
This commit is contained in:
Aaron Orenstein
2025-01-21 11:22:20 -08:00
committed by PyTorch MergeBot
parent 40e27fbcf2
commit 99dbc5b0e2
146 changed files with 801 additions and 1099 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/custom_operator/test_infer_schema_annotation.py
View File

@ -2,7 +2,7 @@
from __future__ import annotations
import typing
from typing import List, Optional, Union
from typing import Optional, Union
import torch
from torch import Tensor, types
@ -87,7 +87,7 @@ class TestInferSchemaWithAnnotation(TestCase):
result = torch.library.infer_schema(foo_op_2, mutates_args=mutates_args)
self.assertEqual(result, "(SymInt[] x) -> SymInt")
def foo_op_3(x: typing.List[int]) -> int:
def foo_op_3(x: list[int]) -> int:
return 1
result = torch.library.infer_schema(foo_op_3, mutates_args=mutates_args)
@ -99,7 +99,7 @@ class TestInferSchemaWithAnnotation(TestCase):
result = torch.library.infer_schema(foo_op_4, mutates_args=mutates_args)
self.assertEqual(result, "(SymInt[]? x) -> SymInt")
def foo_op_5(x: typing.Optional[typing.List[int]]) -> int:
def foo_op_5(x: typing.Optional[list[int]]) -> int:
return 1
result = torch.library.infer_schema(foo_op_5, mutates_args=mutates_args)
@ -136,7 +136,7 @@ class TestInferSchemaWithAnnotation(TestCase):
result = torch.library.infer_schema(foo_op_3, mutates_args=mutates_args)
self.assertEqual(result, "(Tensor x) -> Tensor")
def foo_op_4(x: List[int]) -> types.Number:
def foo_op_4(x: list[int]) -> types.Number:
return x[0]
result = torch.library.infer_schema(foo_op_4, mutates_args=mutates_args)
@ -154,7 +154,7 @@ class TestInferSchemaWithAnnotation(TestCase):
result = torch.library.infer_schema(foo_op_6, mutates_args=mutates_args)
self.assertEqual(result, "(SymInt[] x) -> SymInt")
def foo_op_7(x: List[int]) -> int:
def foo_op_7(x: list[int]) -> int:
return 1
result = torch.library.infer_schema(foo_op_7, mutates_args=mutates_args)
@ -166,7 +166,7 @@ class TestInferSchemaWithAnnotation(TestCase):
result = torch.library.infer_schema(foo_op_8, mutates_args=mutates_args)
self.assertEqual(result, "(SymInt[]? x) -> SymInt")
def foo_op_9(x: Optional[List[int]]) -> int:
def foo_op_9(x: Optional[list[int]]) -> int:
return 1
result = torch.library.infer_schema(foo_op_9, mutates_args=mutates_args)