fix parameterized.expand order

huggingface · Dec 7, 2023 · 76a1e17 · 76a1e17
1 parent 9e133c9
commit 76a1e17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -3130,9 +3130,9 @@ def test_flash_attn_2_generate_padding_right(self):
 
                 self.assertTrue(torch.allclose(out, out_fa))
 
+    @parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
     @require_torch_sdpa
     @slow
-    @parameterized.expand([("float16",), ("bfloat16",), ("float32",)])
     def test_eager_matches_sdpa_inference(self, torch_dtype: str):
         if not self.all_model_classes[0]._supports_sdpa:
             self.skipTest(f"{self.all_model_classes[0].__name__} does not support SDPA")