Skip to content

Commit fc76991

Browse files
committed
only support sequence for s and axes
1 parent aa384d1 commit fc76991

File tree

2 files changed

+0
-14
lines changed

2 files changed

+0
-14
lines changed

dpnp/fft/dpnp_utils_fft.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -438,9 +438,6 @@ def _validate_s_axes(a, s, axes):
438438
if isinstance(s, Sequence):
439439
if any(not isinstance(s_i, int) for s_i in s):
440440
raise_error = True
441-
elif dpnp.is_supported_array_type(s):
442-
if s.ndim != 1 or not dpnp.issubdtype(s, dpnp.integer):
443-
raise_error = True
444441
else:
445442
raise_error = True
446443

tests/test_fft.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -566,17 +566,6 @@ def test_fft_error(self, xp):
566566
# s is not a sequence
567567
assert_raises(TypeError, xp.fft.fftn, a, s=5, axes=(0,))
568568

569-
# s is not a sequence of ints
570-
# dpnp and stock NumPy raise TypeError
571-
# Intel® NumPy raises ValueError
572-
assert_raises(
573-
(TypeError, ValueError),
574-
xp.fft.fftn,
575-
a,
576-
s=xp.array([5.0]),
577-
axes=(0,),
578-
)
579-
580569
# Invalid number of FFT point, invalid s value
581570
assert_raises(ValueError, xp.fft.fftn, a, s=(-5,), axes=(0,))
582571

0 commit comments

Comments
 (0)