Error when rfftn is called with axes and norm='ortho'

[frankong]

I have been getting the following error when rfftn is called with axes and norm='ortho':

~/anaconda3/lib/python3.6/site-packages/mkl_fft/_numpy_fft.py in <listcomp>(.0)
   1040         a = asarray(a)
   1041         s, axes = _cook_nd_args(a, s, axes)
-> 1042         n_tot = numpy.prod([ s[ai] for ai in axes])
   1043 
   1044     output = mkl_fft.rfftn_numpy(a, s, axes)

IndexError: list index out of range

Here is a code snippet to reproduce this:

import numpy as np
x = np.zeros([2, 3])
y = np.fft.rfftn(x, axes=[1], norm='ortho')

numpy without mkl_fft does not produce errors.

[oleksandr-pavlyk]

Thank you @frankong. I will investigate later today.

[oleksandr-pavlyk]

1.0.5 has been released on github. Working to release on conda-forge.