diff options
| author | Bas van Beek <b.f.van.beek@vu.nl> | 2021-04-15 14:40:21 +0200 |
|---|---|---|
| committer | Bas van Beek <b.f.van.beek@vu.nl> | 2021-04-15 14:45:50 +0200 |
| commit | cfbcd38558045197eded5cb84b8c8fe006a77480 (patch) | |
| tree | 2e9ed21200c1357c85354406d7134036a3c3d73e /numpy/fft | |
| parent | 181f273a59744d58f90f45d953a3285484c72cba (diff) | |
| download | numpy-cfbcd38558045197eded5cb84b8c8fe006a77480.tar.gz | |
ENH: Add `__all__` to a number of public modules
Diffstat (limited to 'numpy/fft')
| -rw-r--r-- | numpy/fft/__init__.py | 4 | ||||
| -rw-r--r-- | numpy/fft/__init__.pyi | 4 |
2 files changed, 7 insertions, 1 deletions
diff --git a/numpy/fft/__init__.py b/numpy/fft/__init__.py index a86bb3ac0..fd5e47580 100644 --- a/numpy/fft/__init__.py +++ b/numpy/fft/__init__.py @@ -200,9 +200,13 @@ For examples, see the various functions. """ +from . import _pocketfft, helper from ._pocketfft import * from .helper import * +__all__ = _pocketfft.__all__.copy() +__all__ += helper.__all__ + from numpy._pytesttester import PytestTester test = PytestTester(__name__) del PytestTester diff --git a/numpy/fft/__init__.pyi b/numpy/fft/__init__.pyi index 45190517f..bb4fae903 100644 --- a/numpy/fft/__init__.pyi +++ b/numpy/fft/__init__.pyi @@ -1,4 +1,6 @@ -from typing import Any +from typing import Any, List + +__all__: List[str] fft: Any ifft: Any |