From 74a13c69ba9858d57bd4936f18f3c3bd4b2d580c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C3=A9tan=20Lepage?=
 <33058747+GaetanLepage@users.noreply.github.com>
Date: Mon, 2 Dec 2024 12:03:23 +0100
Subject: [PATCH] [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.
---
 tests/torch_stft_test.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/torch_stft_test.py b/tests/torch_stft_test.py
index 1d29a51..0c7bb30 100644
--- 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: