File tree Expand file tree Collapse file tree 1 file changed +5
-14
lines changed Expand file tree Collapse file tree 1 file changed +5
-14
lines changed Original file line number Diff line number Diff line change 24
24
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25
25
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26
26
27
- import numpy as np
28
-
29
- # helper functions are added for completing the namespaces
30
- if np .lib .NumpyVersion (np .__version__ ) < "2.0.0" :
31
- # intentionally import from the helper module to avoid circular dependency
32
- # when patching numpy fft module with mkl_fft
33
- from numpy .fft .helper import fftfreq , fftshift , ifftshift , rfftfreq
34
- else :
35
- from numpy .fft ._helper import fftfreq , fftshift , ifftshift , rfftfreq
36
-
37
27
# pylint: disable=no-name-in-module
38
28
from ._numpy_fft import (
39
29
fft ,
67
57
"irfftn" ,
68
58
"hfft" ,
69
59
"ihfft" ,
70
- "fftshift" ,
71
- "ifftshift" ,
72
- "fftfreq" ,
73
- "rfftfreq" ,
74
60
]
61
+
62
+ # Added for completing the namespaces
63
+ from numpy .fft import fftfreq , fftshift , ifftshift , rfftfreq
64
+
65
+ __all__ += ["fftshift" , "ifftshift" , "fftfreq" , "rfftfreq" ]
You can’t perform that action at this time.
0 commit comments