From 04683c0659aacf31a1e1df8aa2e6cf7b447a6f12 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 25 Nov 2021 18:59:33 +0100 Subject: [PATCH] Fix a slow test. (#14527) --- tests/test_pipelines_audio_classification.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_pipelines_audio_classification.py b/tests/test_pipelines_audio_classification.py index f01825dd99..ef2dc26aa5 100644 --- a/tests/test_pipelines_audio_classification.py +++ b/tests/test_pipelines_audio_classification.py @@ -114,12 +114,12 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest audio = np.array(dataset[3]["speech"], dtype=np.float32) output = audio_classifier(audio, top_k=4) self.assertEqual( - nested_simplify(output, decimals=4), + nested_simplify(output, decimals=3), [ - {"score": 0.9809, "label": "go"}, - {"score": 0.0073, "label": "up"}, - {"score": 0.0064, "label": "_unknown_"}, - {"score": 0.0015, "label": "down"}, + {"score": 0.981, "label": "go"}, + {"score": 0.007, "label": "up"}, + {"score": 0.006, "label": "_unknown_"}, + {"score": 0.001, "label": "down"}, ], )