Update model_id key to model in prediction_process.py

2025-06-10 08:53:05 -03:00
parent 881e7232e9
commit ba96ea988e
1 changed files with 1 additions and 1 deletions
--- a/laborious/workflows/sub_workflows/prediction_process.py
+++ b/laborious/workflows/sub_workflows/prediction_process.py
@@ -205,7 +205,7 @@ class PredictionProcess():
                {
                    'schema': schema,
                    'table_name': table_name,
-                    'model_id': model_id
+                    'model': model_id
                },
                retry_policy=retry_policy,
                start_to_close_timeout=timedelta(minutes=1),