try to fix all?

Author: lithomas1

array_api_compat/dask/array/_aliases.py

@@ -7,26 +7,7 @@
 
 import numpy as np
 from numpy import (
-    # Constants
-    e,
-    inf,
-    nan,
-    pi,
-    newaxis,
     # Dtypes
-    bool_ as bool,
-    float32,
-    float64,
-    int8,
-    int16,
-    int32,
-    int64,
-    uint8,
-    uint16,
-    uint32,
-    uint64,
-    complex64,
-    complex128,
     iinfo,
     finfo,
     can_cast,
@@ -173,12 +154,9 @@ def asarray(
 
 common_aliases = [alias for alias in _aliases.__all__ if alias not in _da_unsupported]
 
-__all__ = common_aliases + ['asarray', 'bool', 'acos',
-                            'acosh', 'asin', 'asinh', 'atan', 'atan2',
-                            'atanh', 'bitwise_left_shift', 'bitwise_invert',
-                            'bitwise_right_shift', 'concat', 'pow',
-                            'e', 'inf', 'nan', 'pi', 'newaxis', 'float32', 'float64', 'int8',
-                            'int16', 'int32', 'int64', 'uint8', 'uint16', 'uint32', 'uint64',
-                            'complex64', 'complex128', 'iinfo', 'finfo', 'can_cast', 'result_type']
+__all__ = common_aliases + ['asarray', 'acos',
+                    'acosh', 'asin', 'asinh', 'atan', 'atan2',
+                    'atanh', 'bitwise_left_shift', 'bitwise_invert',
+                    'bitwise_right_shift', 'concat', 'pow', 'iinfo', 'finfo', 'can_cast', 'result_type']
 
-_all_ignore = ['get_xp', 'da', 'partial', 'common_aliases', 'np']
+del get_xp, da, common_aliases, np

array_api_compat/dask/array/fft.py

@@ -1,27 +1,22 @@
 from dask.array.fft import * # noqa: F403
-from numpy.fft import __all__ as fft_all
+# cupy.fft doesn't have __all__. If it is added, replace this with
+#
+# from cupy.fft import __all__ as linalg_all
+_n = {}
+exec('from dask.array.fft import *', _n)
+del _n['__builtins__']
+fft_all = list(_n)
+del _n
 
 from ...common import _fft
 from ..._internal import get_xp
 
 import dask.array as da
 
-# fft = get_xp(da)(_fft.fft)
-# ifft = get_xp(da)(_fft.ifft)
-# fftn = get_xp(da)(_fft.fftn)
-# ifftn = get_xp(da)(_fft.ifftn)
-# rfft = get_xp(da)(_fft.rfft)
-# irfft = get_xp(da)(_fft.irfft)
-# rfftn = get_xp(da)(_fft.rfftn)
-# irfftn = get_xp(da)(_fft.irfftn)
-# hfft = get_xp(da)(_fft.hfft)
-# ihfft = get_xp(da)(_fft.ihfft)
 fftfreq = get_xp(da)(_fft.fftfreq)
 rfftfreq = get_xp(da)(_fft.rfftfreq)
-# fftshift = get_xp(da)(_fft.fftshift)
-# ifftshift = get_xp(da)(_fft.ifftshift)
 
-__all__ = fft_all + _fft.__all__
+__all__ = [elem for elem in fft_all if elem != "annotations"] + ["fftfreq", "rfftfreq"]
 
 del get_xp
 del da