Skip to content

Commit 6b80096

Browse files
authored
Skip onnx test if onnxruntime not found. (#2099)
1 parent 66fed64 commit 6b80096

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

ci_test/integration_tests/test_integration_onnx_output.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,12 @@
1515
onnx_model = current_dir + '/experiments/test_integration_onnx_output/lbann.onnx'
1616
import tools
1717
import data.mnist
18-
import onnxruntime
18+
19+
try:
20+
import onnxruntime
21+
except ModuleNotFoundError:
22+
pytest.skip("Skipping ONNX runtime test; onnxruntime not found.",
23+
allow_module_level=True)
1924

2025
# ==============================================
2126
# Options

0 commit comments

Comments
 (0)