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

This commit is contained in:
Joao Gante
2025-02-21 10:18:20 +00:00
committed by GitHub
parent a957b7911a
commit 678885bbbd
39 changed files with 180 additions and 68 deletions

View File

@@ -326,6 +326,8 @@ class IdeficsModelTester:
@require_torch
class IdeficsModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
all_model_classes = (IdeficsModel, IdeficsForVisionText2Text) if is_torch_available() else ()
# Doesn't run generation tests here -- idefics has a dedicated tester for generation tests below
all_generative_model_classes = ()
pipeline_model_mapping = (
{"feature-extraction": IdeficsModel, "image-text-to-text": IdeficsForVisionText2Text}
if is_torch_available()
@@ -868,6 +870,12 @@ class IdeficsForVisionText2TextTest(IdeficsModelTest, GenerationTesterMixin, uni
def test_sdpa_can_dispatch_non_composite_models(self):
pass
@unittest.skip(
"Idefics has a separate test runner for generation tests with complex inheritance, causing this check to fail"
)
def test_generation_tester_mixin_inheritance(self):
pass
@require_torch
@require_vision