Merge pull request #8333 from ev-br/fft__all__

MAINT: fft, linalg: add __all__ lists
cupy · May 29, 2024 · 1fd765f · 1fd765f
2 parents 10e94bd + f7257ac
commit 1fd765f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/cupy/fft/__init__.py b/cupy/fft/__init__.py
@@ -17,3 +17,8 @@
 from cupy.fft._fft import rfftfreq  # NOQA
 from cupy.fft._fft import rfftn  # NOQA
 from cupy.fft import config  # NOQA
+
+__all__ = ["fft", "fft2", "fftfreq", "fftn", "fftshift", "hfft",
+           "ifft", "ifft2", "ifftn", "ifftshift", "ihfft",
+           "irfft", "irfft2", "irfftn",
+           "rfft", "rfft2", "rfftfreq", "rfftn", "config"]
diff --git a/cupy/linalg/__init__.py b/cupy/linalg/__init__.py
@@ -41,3 +41,8 @@
 # Exceptions
 # -----------------------------------------------------------------------------
 from numpy.linalg import LinAlgError  # NOQA
+
+
+__all__ = ["matrix_power", "cholesky", "qr", "svd", "eigh", "eigvalsh", "norm",
+           "det", "matrix_rank", "slogdet", "solve", "tensorsolve", "inv",
+           "pinv", "tensorinv", "LinAlgError"]