Skip to content

Commit ef2d94b

Browse files
committed
fix
1 parent 509cfe4 commit ef2d94b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

_unittests/ut_export/test_issue_2025.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
import unittest
22
import numpy as np
33
import torch
4-
from onnx_diagnostic.ext_test_case import ExtTestCase
4+
from onnx_diagnostic.ext_test_case import ExtTestCase, requires_torch
55

66

77
class TestIssues2025(ExtTestCase):
8+
@requires_torch("2.8")
89
def test_issue_158786_qwen2vl(self):
910
# https://github.com/pytorch/pytorch/issues/158786
1011
class Model(torch.nn.Module):

0 commit comments

Comments
 (0)