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,