fix

interpretml · Dec 26, 2024 · 2e31e1e · 2e31e1e
1 parent 45ba48b
commit 2e31e1e
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