fix

Author: xadupre

.github/workflows/ci.yml

@@ -94,13 +94,16 @@ jobs:
       - name: run tests bypassed
         run: PYTHONPATH=. python _unittests/ut_torch_models/test_tiny_llms_bypassed.py
 
+      - name: test image_classification
+        run: PYTHONPATH=. python _unittests/ut_tasks/test_tasks_image_classification.py
+
       - name: test zero_shot_image_classification
         run: PYTHONPATH=. python _unittests/ut_tasks/test_tasks_zero_shot_image_classification.py
 
       - name: run tests
         run: |
           pip install pytest
-          PYTHONPATH=. UNITTEST_GOING=1 pytest --durations=10 _unittests --ignore _unittests/ut_reference/test_backend_extended_reference_evaluator.py --ignore _unittests/ut_reference/test_backend_onnxruntime_evaluator.py --ignore _unittests/ut_torch_models/test_tiny_llms_bypassed.py --ignore _unittests/ut_tasks/test_tasks_zero_shot_image_classification.py
+          PYTHONPATH=. UNITTEST_GOING=1 pytest --durations=10 _unittests --ignore _unittests/ut_reference/test_backend_extended_reference_evaluator.py --ignore _unittests/ut_reference/test_backend_onnxruntime_evaluator.py --ignore _unittests/ut_torch_models/test_tiny_llms_bypassed.py --ignore _unittests/ut_tasks/test_tasks_zero_shot_image_classification.py --ignore _unittests/ut_tasks/test_tasks_image_classification.py
 
       - name: run backend tests python
         run: PYTHONPATH=. UNITTEST_GOING=1 pytest --durations=10 _unittests/ut_reference/test_backend_extended_reference_evaluator.py

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

@@ -52,7 +52,7 @@ class patched_AttentionMaskConverter:
     """
 
     # This method was fixed in 4.51 at least.
-    _PATCHES_ = ["_make_causal_mask"] if not has_transformers("4.50.9999") else []
+    _PATCHES_ = ["_make_causal_mask"] if not has_transformers("4.48.3") else []
     _PATCHED_CLASS_ = AttentionMaskConverter
 
     @staticmethod
@@ -71,6 +71,9 @@ def _make_causal_mask(
         This static method may be called with ``AttentionMaskConverter._make_causal_mask``
         or ``self._make_causal_mask``. That changes this argument is receives.
         That should not matter but...
+        The patch should be implemented in another way. static methods do not play well
+        with a simple replacement.
+        Fortunately, this patch does not seem to be needed anymore with transformers>=4.48.3.
         """
         if args:
             index = 0 if isinstance(args[0], (tuple, torch.Size)) else 1