gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1712

Browse Source 
Refactor prediction process to use execute_activity_method for MLFlow model transformation and prediction requests, enhancing consistency in workflow execution.
This commit is contained in:
vitor-aignosi
2026-03-26 14:56:50 -03:00
parent ff2999dc14
commit 8a0bb70fc8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
laborious/workflows/sub_workflows/prediction_process.py
View File

@@ -146,7 +146,7 @@ class PredictionProcess:
return return
# Request MLFlow model transformation # Request MLFlow model transformation
transformed_data = await workflow.execute_local_activity_method( transformed_data = await workflow.execute_activity_method(
Activities.request_transform, Activities.request_transform,
{**metadata, 'data': data, 'model_name': model_name, 'model_config': model_config}, {**metadata, 'data': data, 'model_name': model_name, 'model_config': model_config},
retry_policy=retry_policy, retry_policy=retry_policy,
@@ -191,7 +191,7 @@ class PredictionProcess:
): ):
return return
predicted_data = await workflow.execute_local_activity_method( predicted_data = await workflow.execute_activity_method(
Activities.request_predict, Activities.request_predict,
{ {
**metadata, **metadata,