Fix ModelBuilder bug

h2oai · Nov 27, 2024 · 3661e7e · 3661e7e
1 parent bec22a2
commit 3661e7e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/h2o-core/src/main/java/hex/ModelBuilder.java b/h2o-core/src/main/java/hex/ModelBuilder.java
@@ -1236,8 +1236,8 @@ public int separateFeatureVecs() {
         ++res;
       }
     } else {
-      _treatment = null;
-      assert(!hasTreatmentCol());
+      _id = null;
+      assert(!hasIdCol());
     }
     if(isSupervised() && _parms._response_column != null) {
       _response = _train.remove(_parms._response_column);