Fix random seed for flaky test_griffinlim test (#388)

closes #382

Fix random seed for flaky test_griffinlim test (#388)
closes #382
479e666b · Karl Ostmo · Vincent QB · e0f261f3 · 479e666b
Commit 479e666b authored Jan 02, 2020 by Karl Ostmo Committed by Vincent QB Jan 02, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

test/test_functional.py test/test_functional.py +4 -0

No files found.
--- a/test/test_functional.py
+++ b/test/test_functional.py
@@ -71,6 +71,10 @@ class TestFunctional(unittest.TestCase):

    @unittest.skipIf(not IMPORT_LIBROSA, 'Librosa not available')
    def test_griffinlim(self):
+
+        # NOTE: This test is flaky without a fixed random seed
+        # See https://github.com/pytorch/audio/issues/382
+        torch.random.manual_seed(42)
        tensor = torch.rand((1, 1000))

        n_fft = 400