From 496e323115a5bc860954e6b08ce9b2fbdb951f22 Mon Sep 17 00:00:00 2001 From: connorschwartz <46463980+connorschwartz@users.noreply.github.com> Date: Sat, 3 Jan 2026 11:55:21 -0500 Subject: [PATCH] Update benchmark.py Ensemble results need assigned to a variable so that they can be written to a file in the following lines --- openavmkit/benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openavmkit/benchmark.py b/openavmkit/benchmark.py index 348d938..2f1f73b 100644 --- a/openavmkit/benchmark.py +++ b/openavmkit/benchmark.py @@ -3590,7 +3590,7 @@ def _run_hedonic_models( ) if run_ensemble: - _perform_ensemble( + ensemble_results = _perform_ensemble( df_sales=df_sales, df_universe=df_universe, model_group=model_group, @@ -4698,4 +4698,4 @@ def _quick_shap( shaps = _calc_shap(smr.model, X_train, X_train) if plot: - plot_full_beeswarm(shaps, title=title) \ No newline at end of file + plot_full_beeswarm(shaps, title=title)