Don't redirect stderr to stdout

1 files changed, 2 insertions, 3 deletions

diff --git a/config-model/src/main/java/com/yahoo/vespa/model/ml/OnnxModelProbe.java b/config-model/src/main/java/com/yahoo/vespa/model/ml/OnnxModelProbe.java
index 2ef81e3f1fa..2e2ebdeb98f 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/ml/OnnxModelProbe.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/ml/OnnxModelProbe.java
@@ -48,7 +48,7 @@ public class OnnxModelProbe {
                 }
             }
 
-        } catch (IOException | InterruptedException e) {
+        } catch (IllegalArgumentException | IOException | InterruptedException e) {
             e.printStackTrace(System.err);
         }
 
@@ -127,7 +127,6 @@ public class OnnxModelProbe {
 
     private static String callVespaAnalyzeOnnxModel(String jsonInput) throws IOException, InterruptedException {
         ProcessBuilder processBuilder = new ProcessBuilder(binary, "--probe-types");
-        processBuilder.redirectErrorStream(true);
         StringBuilder output = new StringBuilder();
         Process process = processBuilder.start();
 
@@ -136,7 +135,7 @@ public class OnnxModelProbe {
         os.write(jsonInput.getBytes(StandardCharsets.UTF_8));
         os.close();
 
-        // Read output from stdout/stderr
+        // Read output from stdout
         InputStream inputStream = process.getInputStream();
         while (true) {
             int b = inputStream.read();