Skip to content

Commit fba0262

Browse files
authored
use the qualified name as the cache key (#2719)
1 parent 38d7713 commit fba0262

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

gpytorch/utils/memoize.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def _cached(method=None, name=None):
5555

5656
@functools.wraps(method)
5757
def g(self, *args, **kwargs):
58-
cache_name = name if name is not None else method
58+
cache_name = name if name is not None else method.__qualname__
5959
kwargs_pkl = pickle.dumps(kwargs)
6060
if not _is_in_cache(self, cache_name, *args, kwargs_pkl=kwargs_pkl):
6161
return _add_to_cache(self, cache_name, method(self, *args, **kwargs), *args, kwargs_pkl=kwargs_pkl)
@@ -73,7 +73,7 @@ def _cached_ignore_args(method=None, name=None):
7373

7474
@functools.wraps(method)
7575
def g(self, *args, **kwargs):
76-
cache_name = name if name is not None else method
76+
cache_name = name if name is not None else method.__qualname__
7777
if not _is_in_cache_ignore_args(self, cache_name):
7878
return _add_to_cache_ignore_args(self, cache_name, method(self, *args, **kwargs))
7979
return _get_from_cache_ignore_args(self, cache_name)

0 commit comments

Comments
 (0)