[CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180)

2025-02-21 10:18:20 +00:00
parent a957b7911a
commit 678885bbbd
39 changed files with 180 additions and 68 deletions
--- a/tests/models/pegasus_x/test_modeling_pegasus_x.py
+++ b/tests/models/pegasus_x/test_modeling_pegasus_x.py
@@ -847,7 +847,7 @@ class PegasusXStandaloneDecoderModelTester:


@require_torch
-class PegasusXStandaloneDecoderModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
+class PegasusXStandaloneDecoderModelTest(ModelTesterMixin, unittest.TestCase):
    all_model_classes = (PegasusXDecoder,) if is_torch_available() else ()
    test_pruning = False
    is_encoder_decoder = False