From 0bca3f99494b7f838868be9b31151203c8abf073 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 24 Nov 2022 15:56:42 +0530 Subject: [PATCH] test pretrained --- tests/test_inference.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_inference.py b/tests/test_inference.py index f727938..8bdcf4e 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -27,3 +27,12 @@ def test_aggregate(): data=rand, window_size=100, total_frames=1000, step_size=100 ) assert agg_rand.shape[-1] == 1000 + + +def test_pretrained(): + from mayavoz.models import Mayamodel + + model = Mayamodel.from_pretrained( + "shahules786/mayavoz-waveunet-valentini-28spk" + ) + _ = model.enhance("tests/data/vctk/clean_testset_wav/p257_166.wav")