Skip tests properly (#31308)

* Skip tests properly * [test_all] * Add 'reason' as kwarg for skipTest * [test_all] Fix up * [test_all]
2024-06-26 21:59:08 +01:00
parent 1f9f57ab4c
commit 1de7dc7403
254 changed files with 1721 additions and 1298 deletions
--- a/tests/models/megatron_gpt2/test_modeling_megatron_gpt2.py
+++ b/tests/models/megatron_gpt2/test_modeling_megatron_gpt2.py
@@ -31,7 +31,7 @@ if is_torch_available():
@require_tokenizers
 class MegatronGPT2IntegrationTest(unittest.TestCase):
    @slow
-    @unittest.skip("Model is not available.")
+    @unittest.skip(reason="Model is not available.")
    def test_inference_no_head(self):
        directory = "nvidia/megatron-gpt2-345m/"
        if "MYDIR" in os.environ: