Byebye test_batching_equivalence's flakiness (#35729)

* fix * fix * skip * better error message --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-01-21 13:11:33 +01:00
parent 78f5ee0217
commit fd8d61fdb2
18 changed files with 92 additions and 50 deletions
--- a/tests/models/mimi/test_modeling_mimi.py
+++ b/tests/models/mimi/test_modeling_mimi.py
@@ -734,10 +734,6 @@ class MimiModelTest(ModelTesterMixin, unittest.TestCase):
    def test_sdpa_can_compile_dynamic(self):
        pass

-    @is_flaky()
-    def test_batching_equivalence(self):
-        super().test_batching_equivalence()
-

 # Copied from transformers.tests.encodec.test_modeling_encodec.normalize
 def normalize(arr):