fix: Cross validation process will always be completed once started -…

… no incomplete models will be returned to the user
microsoft · Sep 30, 2024 · 6be980a · 6be980a
1 parent e5d95f5
commit 6be980a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/flaml/automl/task/generic_task.py b/flaml/automl/task/generic_task.py
@@ -813,8 +813,6 @@ def evaluate_model_CV(
  if is_spark_dataframe:
  X_train.spark.unpersist() # uncache data to free memory
  X_val.spark.unpersist() # uncache data to free memory
- if budget and time.time() - start_time >= budget:
- break
  val_loss, metric = cv_score_agg_func(val_loss_folds, log_metric_folds)
  n = total_fold_num
  pred_time /= n