try to exit with status code 1 on failure

mlcommons · Aug 27, 2024 · 4f4b64e · 4f4b64e
1 parent afc40f2
commit 4f4b64e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/modelbench/run.py b/src/modelbench/run.py
@@ -427,4 +427,7 @@ def clean_text(str):
 
 
 if __name__ == "__main__":
-    cli()
+    try:
+        cli()
+    except Exception as e:
+        raise SystemExit(e)