Skip to content

GitLab

Unverified Commit 04683c06 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fix a slow test. (#14527)

parent d1fd64e7
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
tests/test_pipelines_audio_classification.py
View file @ 04683c06
...@@ -114,12 +114,12 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest ...@@ -114,12 +114,12 @@ class AudioClassificationPipelineTests(unittest.TestCase, metaclass=PipelineTest
audio = np.array(dataset[3]["speech"], dtype=np.float32) audio = np.array(dataset[3]["speech"], dtype=np.float32)
output = audio_classifier(audio, top_k=4) output = audio_classifier(audio, top_k=4)
self.assertEqual( self.assertEqual(
nested_simplify(output, decimals=4), nested_simplify(output, decimals=3),
[ [
{"score": 0.9809, "label": "go"}, {"score": 0.981, "label": "go"},
{"score": 0.0073, "label": "up"}, {"score": 0.007, "label": "up"},
{"score": 0.0064, "label": "_unknown_"}, {"score": 0.006, "label": "_unknown_"},
{"score": 0.0015, "label": "down"}, {"score": 0.001, "label": "down"},
], ],
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment