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
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"},
],
)
......
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