Update model_id key to model in prediction_process.py

This commit is contained in:
vitor-aignosi
2025-06-10 08:53:05 -03:00
parent 881e7232e9
commit ba96ea988e

View File

@@ -205,7 +205,7 @@ class PredictionProcess():
{ {
'schema': schema, 'schema': schema,
'table_name': table_name, 'table_name': table_name,
'model_id': model_id 'model': model_id
}, },
retry_policy=retry_policy, retry_policy=retry_policy,
start_to_close_timeout=timedelta(minutes=1), start_to_close_timeout=timedelta(minutes=1),