From 4b9da16aa69c1aed137d78d4174485b694d1caa8 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Fri, 24 Oct 2025 10:22:48 -0300 Subject: [PATCH] SIENTIAPDE-1314 SIENTIAPDE-1314 Refactor OPC Write Operation in FormatAndExportPrediction Workflow - Changed the variable name for the result of the OPC write operation for clarity. - Extracted prediction and metrics from the OPC write result to improve data handling. --- laborious/activities/opc.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/laborious/activities/opc.py b/laborious/activities/opc.py index 52b56d9..852f701 100644 --- a/laborious/activities/opc.py +++ b/laborious/activities/opc.py @@ -269,7 +269,7 @@ class OPC(BaseActivity): return success, response_times @activity.defn(name='write_opc_data') - async def write_opc_data(self, input_data: dict[str, Any]) -> dict[Any, Any]: + async def write_opc_data(self, input_data: dict[str, Any]) -> tuple[dict[Any, Any], dict[str, dict[str, float | None]]]: """ Write prediction and confidence data to OPC servers. The two writing operations are optional and independent of each other. @@ -314,10 +314,7 @@ class OPC(BaseActivity): metadata, ) - return { - 'data': self.process_confidence(data, success, metadata), - 'metrics': metrics, - } + return self.process_confidence(data, success, metadata), metrics def process_confidence( self, data: DataFrame, success: bool, metadata: dict[str, Any]