mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
086dec3235
Adds suppressions to pyrefly will typecheck clean: https://github.com/pytorch/pytorch/issues/163283 Almost there! Test plan: dmypy restart && python3 scripts/lintrunner.py -a pyrefly check step 1: delete lines in the pyrefly.toml file from the project-excludes field step 2: run pyrefly check step 3: add suppressions, clean up unused suppressions before: https://gist.github.com/maggiemoss/4b3bf2037014e116bc00706a16aef199 after: INFO 0 errors (5,064 ignored) Only four directories left to enable Pull Request resolved: https://github.com/pytorch/pytorch/pull/164877 Approved by: https://github.com/oulgen
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import functools
|
|
from collections.abc import Callable
|
|
from typing import Concatenate, TypeVar
|
|
from typing_extensions import ParamSpec
|
|
|
|
|
|
_P = ParamSpec("_P")
|
|
_T = TypeVar("_T")
|
|
_C = TypeVar("_C")
|
|
|
|
# Sentinel used to indicate that cache lookup failed.
|
|
_cache_sentinel = object()
|
|
|
|
|
|
def cache_method(
|
|
f: Callable[Concatenate[_C, _P], _T],
|
|
) -> Callable[Concatenate[_C, _P], _T]:
|
|
"""
|
|
Like `@functools.cache` but for methods.
|
|
|
|
`@functools.cache` (and similarly `@functools.lru_cache`) shouldn't be used
|
|
on methods because it caches `self`, keeping it alive
|
|
forever. `@cache_method` ignores `self` so won't keep `self` alive (assuming
|
|
no cycles with `self` in the parameters).
|
|
|
|
Footgun warning: This decorator completely ignores self's properties so only
|
|
use it when you know that self is frozen or won't change in a meaningful
|
|
way (such as the wrapped function being pure).
|
|
"""
|
|
cache_name = "_cache_method_" + f.__name__
|
|
|
|
@functools.wraps(f)
|
|
def wrap(self: _C, *args: _P.args, **kwargs: _P.kwargs) -> _T:
|
|
assert not kwargs
|
|
if not (cache := getattr(self, cache_name, None)):
|
|
cache = {}
|
|
setattr(self, cache_name, cache)
|
|
# pyrefly: ignore # unbound-name
|
|
cached_value = cache.get(args, _cache_sentinel)
|
|
if cached_value is not _cache_sentinel:
|
|
return cached_value
|
|
value = f(self, *args, **kwargs)
|
|
# pyrefly: ignore # unbound-name
|
|
cache[args] = value
|
|
return value
|
|
|
|
return wrap
|