From ca4cc944da22ad8e60c92d727ad684b86ffd69ba Mon Sep 17 00:00:00 2001 From: Pablo Alonso Date: Wed, 4 Mar 2026 09:34:58 +0100 Subject: [PATCH 1/2] Fix MAEST test input/output node names --- .../machinelearning/test_tensorflowpredictmaest.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py b/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py index 572d17a99..819d5456e 100644 --- a/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py +++ b/test/src/unittests/machinelearning/test_tensorflowpredictmaest.py @@ -38,8 +38,16 @@ def setUpClass(self): testdata.models_dir, "maest", "discogs-maest-10s-pw-1.pb" ) - self.model30s = TensorflowPredictMAEST(graphFilename=self.graphFilename30s) - self.model10s = TensorflowPredictMAEST(graphFilename=self.graphFilename10s) + self.model30s = TensorflowPredictMAEST( + graphFilename=self.graphFilename30s, + input="serving_default_melspectrogram", + output="StatefulPartitionedCall:0", + ) + self.model10s = TensorflowPredictMAEST( + graphFilename=self.graphFilename10s, + input="serving_default_melspectrogram", + output="StatefulPartitionedCall:0", + ) @classmethod def tearDownClass(self): From 045991aaf66292c51d6c8dd476919d6064f5ad44 Mon Sep 17 00:00:00 2001 From: Pablo Alonso Date: Wed, 4 Mar 2026 09:37:43 +0100 Subject: [PATCH 2/2] Fix TempoCNN test batchSize value According to the algorythm doc a batchSize of 0 or -1 is a valid parameter that runs a single TF session at the end of the stream. Changing the test value to a negative number (which should raise an exception). --- .../unittests/machinelearning/test_tensorflowpredicttempocnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py b/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py index f22d1f141..3bed15bc7 100644 --- a/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py +++ b/test/src/unittests/machinelearning/test_tensorflowpredicttempocnn.py @@ -112,7 +112,7 @@ def testEmptyInput(self): def testInvalidParam(self): model = join(testdata.models_dir, 'tempocnn', 'deeptemp_k16.pb') self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model, - 'batchSize': 0}) + 'batchSize': -2}) self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model, 'input': 'non_existing'}) self.assertConfigureFails(TensorflowPredictTempoCNN(), {'graphFilename': model,