OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[BE] Enable Ruff's Flake8 PYI030 (#111103)

Browse Source 
Enable [unnecessary-literal-union (PYI030)](https://docs.astral.sh/ruff/rules/unnecessary-literal-union/)

Link: #110950
Pull Request resolved: https://github.com/pytorch/pytorch/pull/111103
Approved by: https://github.com/albanD
This commit is contained in:
isdanni 2023-10-12 13:31:59 +00:00 committed by PyTorch MergeBot
parent 68a1219f74
commit 2f53085f3f
3 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -50,7 +50,6 @@ ignore = [
"PYI018", "PYI018",
"PYI019", "PYI019",
"PYI024", "PYI024",
"PYI030",
"PYI034", "PYI034",
"PYI036", "PYI036",
"PYI041", "PYI041",

2
torch/_refs/fft.py
View File

@ -34,7 +34,7 @@ __all__ = [
"ifftshift", "ifftshift",
] ]
NormType = Union[None, Literal["forward"], Literal["backward"], Literal["ortho"]] NormType = Union[None, Literal["forward", "backward", "ortho"]]
_NORM_VALUES = {None, "forward", "backward", "ortho"} _NORM_VALUES = {None, "forward", "backward", "ortho"}
aten = torch._ops.ops.aten aten = torch._ops.ops.aten

10
torchgen/dest/register_dispatch_key.py
View File

@ -222,11 +222,11 @@ def gen_registration_helpers(backend_index: BackendIndex) -> List[str]:
class RegisterDispatchKey: class RegisterDispatchKey:
backend_index: BackendIndex backend_index: BackendIndex
target: Union[ target: Literal[
Literal[Target.ANONYMOUS_DEFINITION], Target.ANONYMOUS_DEFINITION,
Literal[Target.NAMESPACED_DEFINITION], Target.NAMESPACED_DEFINITION,
Literal[Target.NAMESPACED_DECLARATION], Target.NAMESPACED_DECLARATION,
Literal[Target.REGISTRATION], Target.REGISTRATION,
] ]
# Selector object to determine which operators to generate # Selector object to determine which operators to generate