Skip to content

Commit 71b5f84

Browse files
committed
fix annot
1 parent efa0750 commit 71b5f84

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

_unittests/ut_export/test_dynamic_shapes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import unittest
22
import torch
3-
from onnx_diagnostic.ext_test_case import ExtTestCase
3+
from onnx_diagnostic.ext_test_case import ExtTestCase, requires_transformers
44
from onnx_diagnostic.helpers import string_type
55
from onnx_diagnostic.helpers.cache_helper import make_dynamic_cache
66
from onnx_diagnostic.export import ModelInputs, CoupleInputsDynamicShapes
@@ -739,6 +739,7 @@ def test_couple_input_ds_change_dynamic_dimensions_fixed(self):
739739
self.assertEqual((1, 5, 8), new_input["A"].shape)
740740
self.assertEqual((1, 50), new_input["B"].shape)
741741

742+
@requires_transformers("4.51")
742743
def test_dynamic_cache_replace_by_string(self):
743744
n_layers = 2
744745
bsize, nheads, slen, dim = 2, 4, 3, 7

0 commit comments

Comments
 (0)