return empty prediction interval if mojo response frame does not pres…

…ent interval columns
h2oai · May 12, 2023 · 5d552eb · 5d552eb
1 parent 3675a43
commit 5d552eb
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/...src/main/java/ai/h2o/mojos/deploy/common/transform/MojoFrameToScoreResponseConverter.java b/...src/main/java/ai/h2o/mojos/deploy/common/transform/MojoFrameToScoreResponseConverter.java
@@ -94,17 +94,20 @@ private void fillOutputRows(
    */
   private void fillWithPredictionInterval(
       MojoFrame mojoFrame, ScoreRequest scoreRequest, ScoreResponse scoreResponse) {
-    if (Boolean.TRUE.equals(scoreRequest.isRequestPredictionIntervals())
-        && mojoFrame.getNcols() > 1) {
+    if (Boolean.TRUE.equals(scoreRequest.isRequestPredictionIntervals())) {
       if (!supportPredictionInterval) {
         throw new IllegalStateException(
           "Unexpected error, prediction interval should be supported, but actually not");
       }
-      int targetIdx = getTargetColIdx(Arrays.asList(mojoFrame.getColumnNames()));
-      PredictionInterval predictionInterval = new PredictionInterval();
-      predictionInterval.setFields(getPredictionIntervalFields(mojoFrame, targetIdx));
-      predictionInterval.setRows(getPredictionIntervalRows(mojoFrame, targetIdx));
-      scoreResponse.setPredictionIntervals(predictionInterval);
+      if (mojoFrame.getNcols() > 1) {
+        int targetIdx = getTargetColIdx(Arrays.asList(mojoFrame.getColumnNames()));
+        PredictionInterval predictionInterval = new PredictionInterval();
+        predictionInterval.setFields(getPredictionIntervalFields(mojoFrame, targetIdx));
+        predictionInterval.setRows(getPredictionIntervalRows(mojoFrame, targetIdx));
+        scoreResponse.setPredictionIntervals(predictionInterval);
+      } else {
+        scoreResponse.setPredictionIntervals(new PredictionInterval());
+      }
     }
   }