Skip to content

Commit ea65c80

Browse files
committed
fix issues
1 parent 814a5cf commit ea65c80

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

_unittests/ut_tasks/test_tasks.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,11 @@
11
import unittest
22
import torch
3-
from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_transformers
3+
from onnx_diagnostic.ext_test_case import (
4+
ExtTestCase,
5+
hide_stdout,
6+
has_transformers,
7+
requires_transformers,
8+
)
49
from onnx_diagnostic.torch_models.hghub.model_inputs import get_untrained_model_with_inputs
510
from onnx_diagnostic.torch_export_patches import torch_export_patches
611
from onnx_diagnostic.torch_export_patches.patch_inputs import use_dyn_not_str
@@ -150,6 +155,7 @@ def test_feature_extraction_tiny_bart(self):
150155
model, (), kwargs=inputs, dynamic_shapes=use_dyn_not_str(ds), strict=False
151156
)
152157

158+
@requires_transformers("4.51.999")
153159
@hide_stdout()
154160
def test_summarization(self):
155161
mid = "facebook/bart-large-cnn"

0 commit comments

Comments
 (0)