diff --git a/gpytorch/utils/memoize.py b/gpytorch/utils/memoize.py index 767c0deca..1a7c1d890 100644 --- a/gpytorch/utils/memoize.py +++ b/gpytorch/utils/memoize.py @@ -55,7 +55,7 @@ def _cached(method=None, name=None): @functools.wraps(method) def g(self, *args, **kwargs): - cache_name = name if name is not None else method + cache_name = name if name is not None else method.__qualname__ kwargs_pkl = pickle.dumps(kwargs) if not _is_in_cache(self, cache_name, *args, kwargs_pkl=kwargs_pkl): 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): @functools.wraps(method) def g(self, *args, **kwargs): - cache_name = name if name is not None else method + cache_name = name if name is not None else method.__qualname__ if not _is_in_cache_ignore_args(self, cache_name): return _add_to_cache_ignore_args(self, cache_name, method(self, *args, **kwargs)) return _get_from_cache_ignore_args(self, cache_name)