Skip to content

Commit dca8ec4

Browse files
committed
fix test
1 parent 0bc08eb commit dca8ec4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

_unittests/ut_torch_export_patches/test_patch_torch.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
ExtTestCase,
77
requires_torch,
88
requires_transformers,
9+
has_transformers,
910
has_torch,
1011
)
1112
from onnx_diagnostic.helpers.cache_helper import CacheKeyValue, make_dynamic_cache
@@ -361,6 +362,8 @@ def test_export_with_patch_tiny_llm_dim_meta(self):
361362
self.assertEqualArrayAny(expected, got)
362363

363364
with self.subTest(input="no01", backed_size_oblivious=True):
365+
if not has_transformers("4.55"):
366+
raise unittest.SkipIf("test not working with transformers<4.55")
364367
with (
365368
torch.fx.experimental._config.patch(backed_size_oblivious=True),
366369
torch_export_patches(patch_transformers=True),

0 commit comments

Comments
 (0)