Put result in DKV and name it properly (#15821)

h2oai · Oct 13, 2023 · 5aba9e2 · 5aba9e2
1 parent 096f69d
commit 5aba9e2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/h2o-algos/src/main/java/hex/ensemble/StackedEnsembleModel.java b/h2o-algos/src/main/java/hex/ensemble/StackedEnsembleModel.java
@@ -318,8 +318,12 @@ public Frame scoreContributions(Frame frame, Key<Frame> destination_key, Job<Fra
               !ContributionsWithBackgroundFrameTask.enoughMinMemory(numOfUsefulBaseModels() *
                       ContributionsWithBackgroundFrameTask.estimatePerNodeMinimalMemory(frame.numCols(), frame, backgroundFrame))) // or we have no other choice due to memory
         return SplitToChunksApplyCombine.splitApplyCombine(frame, fun, destination_key);
-      else
-        return fun.apply(frame);
+      else {
+        Frame result = fun.apply(frame);
+        result._key = destination_key;
+        DKV.put(result);
+        return result;
+      }
     } finally {
       Log.info("Finished contributions calculation for " + this._key + "...");
     }