Fix a slow test. (huggingface#14527)

Albertobegue · Jan 27, 2022 · f7e859f · f7e859f
1 parent e49fd42
commit f7e859f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_pipelines_audio_classification.py b/tests/test_pipelines_audio_classification.py
@@ -114,12 +114,12 @@ def test_large_model_pt(self):
         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"},
             ],
         )