From 481b8330bf509c23cbd5bd389d2e4fdb5e6314fd Mon Sep 17 00:00:00 2001 From: Felipe Date: Fri, 10 Jul 2026 00:58:55 -0400 Subject: [PATCH] PredictJob. now builds the saved prediction dataset from the full loaded dataset (all original columns) instead of the input-only --- DashAI/back/job/predict_job.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/DashAI/back/job/predict_job.py b/DashAI/back/job/predict_job.py index efbdee0ca..2b1019fad 100644 --- a/DashAI/back/job/predict_job.py +++ b/DashAI/back/job/predict_job.py @@ -422,7 +422,7 @@ def run( manual_input_data, dataset_trained_path ) - prepared_dataset, y_pred = _run_prediction_pipeline( + _, y_pred = _run_prediction_pipeline( task=task, trained_model=trained_model, train_dataset=train_dataset, @@ -460,9 +460,13 @@ def run( full_path = Path(path) / folder_name full_path.mkdir(parents=True, exist_ok=True) - # Add predictions to loaded dataset + output_col = model_session.output_columns[0] + base_columns = [ + col for col in loaded_dataset.column_names if col != output_col + ] + output_dataset = loaded_dataset.select_columns(base_columns) dataset_with_prediction = to_dashai_dataset( - prepared_dataset.add_column(model_session.output_columns[0], y_pred) + output_dataset.add_column(output_col, y_pred) ) # Filter schema from trained dataset