From ba96ea988ef1bdd34eaca0aa54436ce8bd1c6243 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Tue, 10 Jun 2025 08:53:05 -0300
Subject: [PATCH] Update model_id key to model in prediction_process.py

---
 laborious/workflows/sub_workflows/prediction_process.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py
index e4fc8cc..1e01f6a 100644
--- 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),