Fix memory leak that occurs when purification is used to calculate ga…

…in, but not the updates.
interpretml · Dec 26, 2024 · 50ab4ba · 50ab4ba
1 parent 5e999b6
commit 50ab4ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shared/libebm/GenerateTermUpdate.cpp b/shared/libebm/GenerateTermUpdate.cpp
@@ -585,10 +585,10 @@ static ErrorEbm BoostMultiDimensional(BoosterShell* const pBoosterShell,
       }
 
       *pTotalGain = gain;
-
-      free(aWeights);
    }
 
+   free(aWeights);
+
 #ifndef NDEBUG
    free(aDebugCopyBins);
 #endif // NDEBUG