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 66fed64 commit 6b80096Copy full SHA for 6b80096
ci_test/integration_tests/test_integration_onnx_output.py
@@ -15,7 +15,12 @@
15
onnx_model = current_dir + '/experiments/test_integration_onnx_output/lbann.onnx'
16
import tools
17
import data.mnist
18
-import onnxruntime
+
19
+try:
20
+ import onnxruntime
21
+except ModuleNotFoundError:
22
+ pytest.skip("Skipping ONNX runtime test; onnxruntime not found.",
23
+ allow_module_level=True)
24
25
# ==============================================
26
# Options
0 commit comments