diff --git a/src/array_api_compat/torch/_info.py b/src/array_api_compat/torch/_info.py index 050c7846..fba69f00 100644 --- a/src/array_api_compat/torch/_info.py +++ b/src/array_api_compat/torch/_info.py @@ -9,7 +9,6 @@ """ import torch -from functools import cache class __array_namespace_info__: """ @@ -249,7 +248,6 @@ def _dtypes(self, kind): return res raise ValueError(f"unsupported kind: {kind!r}") - @cache def dtypes(self, *, device=None, kind=None): """ The array API data types supported by PyTorch. @@ -313,7 +311,6 @@ def dtypes(self, *, device=None, kind=None): del res[k] return res - @cache def devices(self): """ The devices supported by PyTorch.