requires

Author: xadupre

_unittests/ut_helpers/test_torch_helper.py

@@ -4,7 +4,7 @@
 import onnx
 import torch
 import transformers
-from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout
+from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, requires_torch
 from onnx_diagnostic.helpers import max_diff, string_type
 from onnx_diagnostic.helpers.torch_helper import (
     dummy_llm,
@@ -274,6 +274,7 @@ def test_torch_deepcopy_cache_dce(self):
         self.assertEqual(hash1, hash2)
         self.assertGreater(torch_tensor_size(cc), 1)
 
+    @requires_torch("4.50")
     def test_torch_deepcopy_mamba_cache(self):
         cache = make_mamba_cache(
             [