Remove reference to internals in torch.onnx in test (#17550)

- #11901
microsoft · Sep 19, 2023 · d350ab3 · d350ab3
1 parent f969e7f
commit d350ab3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/onnxruntime/test/python/test_pytorch_export_contrib_ops.py b/onnxruntime/test/python/test_pytorch_export_contrib_ops.py
@@ -49,9 +49,7 @@ def to_numpy(tensor):
 # These set of tests verify ONNX model export and compares outputs between
 # PyTorch and ORT.
 class ONNXExporterTest(unittest.TestCase):
-    from torch.onnx.symbolic_helper import _export_onnx_opset_version
-
-    opset_version = _export_onnx_opset_version
+    opset_version = 17
     keep_initializers_as_inputs = True  # For IR version 3 type export.
 
     def setUp(self):