Skip to content

Commit 2df2cae

Browse files
Corrected utility __supported_array_or_not_implemented
1 parent 70be10e commit 2df2cae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mkl_fft/_float_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from numpy import (half, float32, asarray, ndarray,
2828
longdouble, float64, longcomplex, complex_, float128, complex256)
2929

30-
__all__ = ['__upcast_float16_array', '__downcast_float128_array']
30+
__all__ = ['__upcast_float16_array', '__downcast_float128_array', '__supported_array_or_not_implemented']
3131

3232
def __upcast_float16_array(x):
3333
"""
@@ -81,4 +81,4 @@ def __supported_array_or_not_implemented(x):
8181
__x = asarray(x)
8282
if __x.dtype in [half, float128, complex256]:
8383
return NotImplemented
84-
return x
84+
return __x

0 commit comments

Comments
 (0)