Skip to content

Commit 56c26fe

Browse files
authored
Merge pull request #472 from betatim/remove-caching-for-torch
Remove caching from torch information endpoint
2 parents 54e351c + d087d3c commit 56c26fe

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

src/array_api_compat/torch/_info.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
"""
1010
import torch
1111

12-
from functools import cache
1312

1413
class __array_namespace_info__:
1514
"""
@@ -249,7 +248,6 @@ def _dtypes(self, kind):
249248
return res
250249
raise ValueError(f"unsupported kind: {kind!r}")
251250

252-
@cache
253251
def dtypes(self, *, device=None, kind=None):
254252
"""
255253
The array API data types supported by PyTorch.
@@ -313,7 +311,6 @@ def dtypes(self, *, device=None, kind=None):
313311
del res[k]
314312
return res
315313

316-
@cache
317314
def devices(self):
318315
"""
319316
The devices supported by PyTorch.

0 commit comments

Comments
 (0)