From bab109e076ca9dfd40b21841fcf000a7752214ff Mon Sep 17 00:00:00 2001
From: Veronika Maurerova <veronika.maurerova@h2o.ai>
Date: Wed, 23 Aug 2023 11:23:54 +0200
Subject: [PATCH] rebase master

---
 h2o-genmodel/src/main/java/hex/genmodel/GenModel.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java b/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java
index f5fc63e25a52..dd8298829c36 100755
--- a/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java
+++ b/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java
@@ -150,7 +150,7 @@ public String[] getOutputNames() {
         outputNames = new String[]{"lp"};
         break;
       case BinomialUplift:
-        outputNames = new String[]{"uplift_predict", "p_y1_ct1", "p_y1_ct0"};
+        outputNames = new String[]{"uplift_predict", "p_y1_with_treatment", "p_y1_without_treatment"};
         break;
       default:
         throw new UnsupportedOperationException("Getting output column names for model category '" +