update scipy.fftpack file

Author: vtavana

mkl_fft/_numpy_fft.py

@@ -75,7 +75,8 @@
 
 import numpy as np
 
-from . import _pydfti as mkl_fft  # pylint: disable=no-name-in-module
+import mkl_fft
+
 from ._fft_utils import check_norm, compute_fwd_scale
 from ._float_utils import __downcast_float128_array
 

mkl_fft/_scipy_fft.py

@@ -32,7 +32,8 @@
 import mkl
 import numpy as np
 
-from . import _pydfti as mkl_fft  # pylint: disable=no-name-in-module
+import mkl_fft
+
 from ._fft_utils import compute_fwd_scale
 from ._float_utils import __supported_array_or_not_implemented
 

mkl_fft/_scipy_fftpack.py

@@ -24,47 +24,48 @@
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-from . import _float_utils
-from . import _pydfti as mkl_fft  # pylint: disable=no-name-in-module
+import mkl_fft
+
+from ._float_utils import __upcast_float16_array
 
 __all__ = ["fft", "ifft", "fftn", "ifftn", "fft2", "ifft2", "rfft", "irfft"]
 
 
 def fft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.fft(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def ifft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.ifft(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def fftn(a, shape=None, axes=None, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.fftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.fftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def ifftn(a, shape=None, axes=None, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.ifftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.ifftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def fft2(a, shape=None, axes=(-2, -1), overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.fftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.fftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def ifft2(a, shape=None, axes=(-2, -1), overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.ifftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.ifftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def rfft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.rfftpack(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def irfft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.irfftpack(x, n=n, axis=axis, overwrite_x=overwrite_x)