From f76ae839a96a3b5ff75eca3c51746586ff3cee79 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Fri, 27 Jun 2025 10:06:38 -0300
Subject: [PATCH] SIENTIAPDE-1110  Refactor metadata handling in
 PredictionsBatch and PredictionProcess to improve clarity in input
 preparation.

---
 laborious/workflows/predictions_batch.py                | 2 +-
 laborious/workflows/sub_workflows/prediction_process.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/laborious/workflows/predictions_batch.py b/laborious/workflows/predictions_batch.py
index f643c7d..9d3bcd6 100644
--- a/laborious/workflows/predictions_batch.py
+++ b/laborious/workflows/predictions_batch.py
@@ -61,7 +61,7 @@ class PredictionsBatch():
 
         # Prepare input for prediction_process workflow
         prediction_input = {
-            **metadata,
+            'metadata': metadata,
             'data': data,
             'schema': input_data['schema'],
             'table_name': input_data['table_name'],
diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py
index b89fa0c..3aecce2 100644
--- a/laborious/workflows/sub_workflows/prediction_process.py
+++ b/laborious/workflows/sub_workflows/prediction_process.py
@@ -235,7 +235,7 @@ class PredictionProcess():
             await workflow.execute_child_workflow(
                 'format_and_export_prediction',
                 {
-                    **metadata,
+                    'metadata': metadata,
                     'path_flag': path_flag,
                     'data': data,
                     'prediction_confidence': confidence,