try

Author: xadupre

_unittests/ut_torch_models/test_llm_phi2.py

@@ -1,6 +1,11 @@
 import unittest
 import torch
-from onnx_diagnostic.ext_test_case import ExtTestCase, ignore_warnings, requires_transformers
+from onnx_diagnostic.ext_test_case import (
+    ExtTestCase,
+    ignore_warnings,
+    requires_transformers,
+    requires_python,
+)
 from onnx_diagnostic.torch_models.llms import get_phi2
 from onnx_diagnostic.helpers import string_type
 from onnx_diagnostic.torch_export_patches import bypass_export_some_errors
@@ -25,6 +30,7 @@ def test_export_phi2_1(self):
         assert ep
 
     @ignore_warnings(UserWarning)
+    @requires_python((3, 12))
     def test_export_phi2_2_bypassed(self):
         data = get_phi2(num_hidden_layers=2)
         model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]

onnx_diagnostic/ext_test_case.py

@@ -399,6 +399,18 @@ def has_cuda() -> bool:
     return torch.cuda.device_count() > 0
 
 
+def requires_python(version: Tuple[int,...], msg: str = ""):
+    """
+    Skips a test if python is too old.
+
+    :param msg: to overwrite the message
+    :param version: minimum version
+    """
+    if sys.version_info[:len(version)] < version:
+        return unittest.skip(msg or f"python not recent enough {sys.version_info} < {version}")
+    return lambda x: x
+
+
 def requires_cuda(msg: str = "", version: str = "", memory: int = 0):
     """
     Skips a test if cuda is not available.