SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix a slow test. (#14527)

Browse Source
This commit is contained in:
Nicolas Patry
2021-11-25 18:59:33 +01:00
committed by GitHub
parent d1fd64e7aa
commit 04683c0659
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_pipelines_audio_classification.py
View File

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