Skip to content

Commit da6d4e4

Browse files
committed
Fix cupy fft __all__
1 parent 18960aa commit da6d4e4

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

array_api_compat/cupy/fft.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,12 @@
11
from cupy.fft import * # noqa: F403
2-
from cupy.fft import __all__ as fft_all
2+
# cupy.fft doesn't have __all__. If it is added, replace this with
3+
#
4+
# from cupy.fft import __all__ as linalg_all
5+
_n = {}
6+
exec('from cupy.fft import *', _n)
7+
del _n['__builtins__']
8+
fft_all = list(_n)
9+
del _n
310

411
from ..common import _fft
512
from .._internal import get_xp

0 commit comments

Comments
 (0)