mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
6c2a8b6b38
Applies a bunch of new ruff lint rules that are now stable. Some of these improve efficiency or readability. Since I already did passes on the codebase for these when they were in preview, there should be relatively few changes to the codebase. This is just more for future hardening of it. Pull Request resolved: https://github.com/pytorch/pytorch/pull/129825 Approved by: https://github.com/XuehaiPan, https://github.com/jansel, https://github.com/malfet
77 lines
1.6 KiB
Python
77 lines
1.6 KiB
Python
import torch
|
|
|
|
|
|
def check_error(desc, fn, *required_substrings):
|
|
try:
|
|
fn()
|
|
except Exception as e:
|
|
error_message = e.args[0]
|
|
print("=" * 80)
|
|
print(desc)
|
|
print("-" * 80)
|
|
print(error_message)
|
|
print()
|
|
for sub in required_substrings:
|
|
assert sub in error_message
|
|
return
|
|
raise AssertionError(f"given function ({desc}) didn't raise an error")
|
|
|
|
|
|
check_error("Wrong argument types", lambda: torch.FloatStorage(object()), "object")
|
|
|
|
check_error(
|
|
"Unknown keyword argument", lambda: torch.FloatStorage(content=1234.0), "keyword"
|
|
)
|
|
|
|
check_error(
|
|
"Invalid types inside a sequence",
|
|
lambda: torch.FloatStorage(["a", "b"]),
|
|
"list",
|
|
"str",
|
|
)
|
|
|
|
check_error("Invalid size type", lambda: torch.FloatStorage(1.5), "float")
|
|
|
|
check_error(
|
|
"Invalid offset", lambda: torch.FloatStorage(torch.FloatStorage(2), 4), "2", "4"
|
|
)
|
|
|
|
check_error(
|
|
"Negative offset", lambda: torch.FloatStorage(torch.FloatStorage(2), -1), "2", "-1"
|
|
)
|
|
|
|
check_error(
|
|
"Invalid size",
|
|
lambda: torch.FloatStorage(torch.FloatStorage(3), 1, 5),
|
|
"2",
|
|
"1",
|
|
"5",
|
|
)
|
|
|
|
check_error(
|
|
"Negative size",
|
|
lambda: torch.FloatStorage(torch.FloatStorage(3), 1, -5),
|
|
"2",
|
|
"1",
|
|
"-5",
|
|
)
|
|
|
|
check_error("Invalid index type", lambda: torch.FloatStorage(10)["first item"], "str")
|
|
|
|
|
|
def assign():
|
|
torch.FloatStorage(10)[1:-1] = "1"
|
|
|
|
|
|
check_error("Invalid value type", assign, "str")
|
|
|
|
check_error(
|
|
"resize_ with invalid type", lambda: torch.FloatStorage(10).resize_(1.5), "float"
|
|
)
|
|
|
|
check_error(
|
|
"fill_ with invalid type", lambda: torch.IntStorage(10).fill_("asdf"), "str"
|
|
)
|
|
|
|
# TODO: frombuffer
|