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: Add noqa to necessary tests (#146391)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146391
Approved by: https://github.com/justinchuby, https://github.com/Skylion007
This commit is contained in:
Aaron Orenstein
2025-02-06 07:23:33 -08:00
committed by PyTorch MergeBot
parent b61032fcf7
commit 1f8ff94d4f
7 changed files with 63 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
test/test_fx_experimental.py
View File

@ -12,7 +12,7 @@ import tempfile
import typing
import unittest
from types import BuiltinFunctionType
from typing import Callable, List, NamedTuple, Optional, Union
from typing import Callable, NamedTuple, Optional, Union
import torch
import torch.fx.experimental.meta_tracer
@ -1548,25 +1548,25 @@ class {test_classname}(torch.nn.Module):
(Optional[list[int]], list[int]),
] + [
# pre-PEP585 signatures
(typing.List[int], int),
(typing.List[int], create_type_hint([int, int])),
(typing.List[int], create_type_hint((int, int))),
(typing.List[torch.Tensor], create_type_hint([torch.Tensor, torch.Tensor])),
(typing.List[int], int), # noqa: UP006
(typing.List[int], create_type_hint([int, int])), # noqa: UP006
(typing.List[int], create_type_hint((int, int))), # noqa: UP006
(typing.List[torch.Tensor], create_type_hint([torch.Tensor, torch.Tensor])), # noqa: UP006
(
typing.List[torch.Tensor],
typing.List[torch.Tensor], # noqa: UP006
create_type_hint([torch.nn.Parameter, torch.nn.Parameter]),
),
(typing.List[torch.Tensor], create_type_hint([torch.nn.Parameter, torch.Tensor])),
(typing.List[torch.Tensor], create_type_hint([torch.Tensor, torch.nn.Parameter])),
(typing.List[torch.Tensor], create_type_hint((torch.Tensor, torch.Tensor))),
(typing.List[torch.Tensor], create_type_hint([torch.nn.Parameter, torch.Tensor])), # noqa: UP006
(typing.List[torch.Tensor], create_type_hint([torch.Tensor, torch.nn.Parameter])), # noqa: UP006
(typing.List[torch.Tensor], create_type_hint((torch.Tensor, torch.Tensor))), # noqa: UP006
(
typing.List[torch.Tensor],
typing.List[torch.Tensor], # noqa: UP006
create_type_hint((torch.nn.Parameter, torch.nn.Parameter)),
),
(typing.List[torch.Tensor], create_type_hint((torch.nn.Parameter, torch.Tensor))),
(typing.List[torch.Tensor], create_type_hint((torch.Tensor, torch.nn.Parameter))),
(Optional[typing.List[torch.Tensor]], typing.List[torch.Tensor]),
(Optional[typing.List[int]], typing.List[int]),
(typing.List[torch.Tensor], create_type_hint((torch.nn.Parameter, torch.Tensor))), # noqa: UP006
(typing.List[torch.Tensor], create_type_hint((torch.Tensor, torch.nn.Parameter))), # noqa: UP006
(Optional[typing.List[torch.Tensor]], typing.List[torch.Tensor]), # noqa: UP006
(Optional[typing.List[int]], typing.List[int]), # noqa: UP006
]
for sig_type, arg_type in should_be_equal:
@ -1575,7 +1575,7 @@ class {test_classname}(torch.nn.Module):
should_fail = [
(int, float),
(Union[int, float], str),
(list[torch.Tensor], List[int]),
(list[torch.Tensor], typing.List[int]), # noqa: UP006
] + [
# pre-PEP585 signatures
(list[torch.Tensor], list[int]),