Catch KeyboardInterrupt (#23)

justinchuby · May 19, 2024 · 1733a93 · 1733a93
1 parent 6be2167
commit 1733a93
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/model_explorer_onnx/bin/onnxvis.py b/src/model_explorer_onnx/bin/onnxvis.py
@@ -7,14 +7,17 @@
 
 def main():
     # Run model explorer
-    subprocess.run(
-        [
-            "model-explorer",
-            "--extensions",
-            "model_explorer_onnx",
-            *sys.argv[1:],
-        ]
-    )
+    try:
+        subprocess.run(
+            [
+                "model-explorer",
+                "--extensions",
+                "model_explorer_onnx",
+                *sys.argv[1:],
+            ]
+        )
+    except KeyboardInterrupt:
+        pass
 
 
 if __name__ == "__main__":