From 71ca3a7648f501dd2c508ca9da5d520823e935c3 Mon Sep 17 00:00:00 2001
From: Paul Koch <code@koch.ninja>
Date: Wed, 25 Dec 2024 13:07:55 -0800
Subject: [PATCH] zeroed tensor at end of Flatten

---
 shared/libebm/PartitionOneDimensionalBoosting.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/shared/libebm/PartitionOneDimensionalBoosting.cpp b/shared/libebm/PartitionOneDimensionalBoosting.cpp
index 63a875705..12a5f0649 100644
--- a/shared/libebm/PartitionOneDimensionalBoosting.cpp
+++ b/shared/libebm/PartitionOneDimensionalBoosting.cpp
@@ -350,8 +350,10 @@ static ErrorEbm Flatten(BoosterShell* const pBoosterShell,
 
                EBM_ASSERT(bNominal || pSplit == cSlices - 1 + pInnerTermUpdate->GetSplitPointer(iDimension));
 
+               pInnerTermUpdate->SetCountSlices(iDimension, 1);
+
                LOG_0(Trace_Verbose, "Exited Flatten");
-               return Error_IllegalParamVal;
+               return Error_None;
             }
             if(!pTreeNode->DECONSTRUCT_IsRightChildTraversal()) {
                // we checked earlier that countBins could be converted to a UIntSplit