[tests] provide return_complex to torch.stft (#23)

As per https://pytorch.org/docs/stable/generated/torch.stft.html, providing return_complex is now mandatory.
asteroid-team · Dec 2, 2024 · 74a13c6 · 74a13c6
1 parent f3142cb
commit 74a13c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/torch_stft_test.py b/tests/torch_stft_test.py
@@ -128,6 +128,7 @@ def test_torch_stft(
         pad_mode=pad_mode,
         normalized=normalized,
         onesided=True,
+        return_complex=True,
     )
 
     spec_asteroid = stft(wav)
@@ -145,6 +146,7 @@ def test_torch_stft(
             normalized=normalized,
             onesided=True,
             length=output_len,
+            return_complex=True,
         )
 
     except RuntimeError: