0.6.16 a68

lewuyou · Oct 7, 2024 · c11f588 · c11f588
1 parent 0da8728
commit c11f588
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autots/evaluator/auto_model.py b/autots/evaluator/auto_model.py
@@ -1645,7 +1645,7 @@ def _eval_prediction_for_template(
         )
         template_result.full_mae_ids.extend([model_id])
         template_result.full_mae_vals.extend([validation_round])
-    return template_result
+    return template_result, best_smape
 
 
 horizontal_post_processors = [
@@ -1826,7 +1826,7 @@ def virtual_memory():
             else:
                 post_memory_percent = 0.0
             # wrapped up to enable postprocessing
-            template_result = _eval_prediction_for_template(
+            template_result, best_smape = _eval_prediction_for_template(
                     df_forecast, template_result, verbose,
                     actuals, weights, df_trn_arr, ensemble,
                     scaler, cumsum_A, diff_A, last_of_array,
@@ -1858,7 +1858,7 @@ def virtual_memory():
                     )
                     df_forecast2.transformation_parameters = x
                     template_result.model_count += 1
-                    template_result = _eval_prediction_for_template(
+                    template_result, best_smape = _eval_prediction_for_template(
                             df_forecast2, template_result, verbose,
                             actuals, weights, df_trn_arr, ensemble,
                             scaler, cumsum_A, diff_A, last_of_array,