From 8a0bb70fc80c2d52280b33531a52b98c407e6362 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Thu, 26 Mar 2026 14:56:50 -0300
Subject: [PATCH] SIENTIAPDE-1712  Refactor prediction process to use
 execute_activity_method for MLFlow model transformation and prediction
 requests, enhancing consistency in workflow execution.

---
 laborious/workflows/sub_workflows/prediction_process.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py
index 3493b59..7c4d702 100644
--- a/laborious/workflows/sub_workflows/prediction_process.py
+++ b/laborious/workflows/sub_workflows/prediction_process.py
@@ -146,7 +146,7 @@ class PredictionProcess:
             return
 
         # Request MLFlow model transformation
-        transformed_data = await workflow.execute_local_activity_method(
+        transformed_data = await workflow.execute_activity_method(
             Activities.request_transform,
             {**metadata, 'data': data, 'model_name': model_name, 'model_config': model_config},
             retry_policy=retry_policy,
@@ -191,7 +191,7 @@ class PredictionProcess:
         ):
             return
 
-        predicted_data = await workflow.execute_local_activity_method(
+        predicted_data = await workflow.execute_activity_method(
             Activities.request_predict,
             {
                 **metadata,