add test for onnx import

Signed-off-by: Alexandre Eichenberger <alexe@us.ibm.com>
onnx · Dec 3, 2024 · 0ac4608 · 0ac4608
1 parent 6939f8e
commit 0ac4608
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/utils/RunONNXModel.py b/utils/RunONNXModel.py
@@ -14,7 +14,11 @@
 import os
 import sys
 import argparse
-import onnx
+try:
+    import onnx
+except ImportError:
+    print("Import of onnx package failed; please instal to use this script")
+    exit(1)
 import time
 import signal
 import subprocess