Skip to content

Commit a3def7a

Browse files
committed
fix
1 parent 1963e27 commit a3def7a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

_unittests/ut_export/test_control_flow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
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_torch
44
from onnx_diagnostic.export.control_flow import loop_for
55

66

77
class TestControlFlow(ExtTestCase):
8+
@requires_torch("2.9.99")
89
def test_loop_for(self):
910
class Model(torch.nn.Module):
1011
def forward(self, n_iter, x):

0 commit comments

Comments
 (0)