Skip to content

Commit fe0ad05

Browse files
author
Alejandro Gaston Alvarez Franceschi
committed
Remove ISTFT test class and complex parameter
1 parent 2605f55 commit fe0ad05

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

coremltools/converters/mil/frontend/torch/test/test_torch_ops.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9587,10 +9587,9 @@ def forward(self, x):
95879587
compute_unit=compute_unit
95889588
)
95899589

9590-
class TestISTFT(TorchBaseTest):
95919590
@pytest.mark.slow
95929591
@pytest.mark.parametrize(
9593-
"compute_unit, backend, input_shape, complex, n_fft, hop_length, win_length, window, center, pad_mode, normalized, onesided, length",
9592+
"compute_unit, backend, input_shape, n_fft, hop_length, win_length, window, center, normalized, onesided, length, return_complex",
95949593
itertools.product(
95959594
compute_units,
95969595
backends,

0 commit comments

Comments
 (0)