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

More ruff SIM fixes (#164695)

Browse Source 
This PR applies ruff `SIM` rules to more files. Most changes are about simplifying `dict.get` because `None` is already the default value.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164695
Approved by: https://github.com/ezyang
This commit is contained in:
Yuanyuan Chen
2025-10-09 03:24:46 +00:00
committed by PyTorch MergeBot
parent 54ae61c573
commit a029675f6f
49 changed files with 153 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
test/test_jit_string.py
View File

@ -241,17 +241,17 @@ class TestScript(JitTestCase):
def test_split() -> tuple[list[str], list[str], list[str], list[str], list[str],
list[str], list[str], list[str], list[str], list[str], list[str]]:
return (
"a a a a a".split(),
"a a a a a".split(),
" a a\ta \v a \v\f\n a \t ".split(),
" a a a a a ".split(" "),
"a a a a a ".split(" ", 10),
"a a a a a ".split(" ", -1),
"a a a a a ".split(" ", 3),
" a a a a a ".split("*"),
" a*a a*a a".split("*"),
" a*a a*a a ".split("*", -1),
" a*a a*a a ".split("a*", 10),
["a", "a", "a", "a", "a"],
["a", "a", "a", "a", "a"],
["a", "a", "a", "a", "a"],
["", "a", "a", "a", "a", "a", ""],
["a", "a", "a", "a", "a", ""],
["a", "a", "a", "a", "a", ""],
["a", "a", "a", "a a "],
[" a a a a a "],
[" a", "a a", "a a"],
[" a", "a a", "a a "],
[" ", "a ", "a a "],
)
self.checkScript(test_split, ())
@ -266,15 +266,15 @@ class TestScript(JitTestCase):
def test_rsplit() -> tuple[list[str], list[str], list[str], list[str], list[str],
list[str], list[str], list[str], list[str]]:
return (
"a a a a a".rsplit(),
" a a a a a ".rsplit(" "),
"a a a a a ".rsplit(" ", 10),
"a a a a a ".rsplit(" ", -1),
"a a a a a ".rsplit(" ", 3),
" a a a a a ".rsplit("*"),
" a*a a*a a ".rsplit("*"),
" a*a a*a a ".rsplit("*", -1),
" a*a a*a a".rsplit("a*", 10),
["a", "a", "a", "a", "a"],
["", "a", "a", "a", "a", "a", ""],
["a", "a", "a", "a", "a", ""],
["a", "a", "a", "a", "a", ""],
["a a a", "a", "a", ""],
[" a a a a a "],
[" a", "a a", "a a "],
[" a", "a a", "a a "],
[" ", "a ", "a a"],
)
self.checkScript(test_rsplit, ())