make the err msg short

h2oai · Sep 1, 2024 · 70ded53 · 70ded53
1 parent 9a8df8d
commit 70ded53
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/h2o-core/src/main/java/water/rapids/ast/prims/internal/AstRunTool.java b/h2o-core/src/main/java/water/rapids/ast/prims/internal/AstRunTool.java
@@ -36,7 +36,9 @@ public ValStr apply(Env env, Env.StackHelp stk, AstRoot[] asts) {
             Method mainMethod = clazz.getDeclaredMethod("mainInternal", String[].class);
             mainMethod.invoke(null, new Object[]{args});
         } catch (Exception e) {
-            throw new RuntimeException(e);
+            RuntimeException exceptionWithShortStacktrace = new RuntimeException(e.getMessage());
+            exceptionWithShortStacktrace.setStackTrace(new StackTraceElement[]{e.getStackTrace()[0]});
+            throw exceptionWithShortStacktrace;
         }
         return new ValStr("OK");
     }

diff --git a/h2o-extensions/xgboost/src/main/java/water/tools/XGBoostLibExtractTool.java b/h2o-extensions/xgboost/src/main/java/water/tools/XGBoostLibExtractTool.java
@@ -13,7 +13,7 @@ public static void main(String[] args) throws IOException {
         try {
             mainInternal(args);
         } catch (IllegalArgumentException e) {
-            System.err.println((e.getMessage());
+            System.err.println((e.getMessage()));
             System.exit(1);
         }
     }