We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1963e27 commit a3def7aCopy full SHA for a3def7a
_unittests/ut_export/test_control_flow.py
@@ -1,10 +1,11 @@
1
import unittest
2
import torch
3
-from onnx_diagnostic.ext_test_case import ExtTestCase
+from onnx_diagnostic.ext_test_case import ExtTestCase, requires_torch
4
from onnx_diagnostic.export.control_flow import loop_for
5
6
7
class TestControlFlow(ExtTestCase):
8
+ @requires_torch("2.9.99")
9
def test_loop_for(self):
10
class Model(torch.nn.Module):
11
def forward(self, n_iter, x):
0 commit comments