From 7f3457e318669acdad9fd94600e1da79eb7dfc50 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Thu, 14 Aug 2025 15:41:03 -0300 Subject: [PATCH] SIENTIAPDE-1169 Enhance BaseActivity initialization across multiple activities to include error counter - Updated the initialization of the BaseActivity in Gates, MLFlow, and OPC classes to set the error counter to True, improving error tracking and handling capabilities. --- laborious/activities/gates.py | 3 ++- laborious/activities/mlflow.py | 3 ++- laborious/activities/opc.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py index b958566..dbc1396 100644 --- a/laborious/activities/gates.py +++ b/laborious/activities/gates.py @@ -48,7 +48,8 @@ mlflow_content_filter_functions = { class Gates(BaseActivity): def __init__(self, logger: Logger, notification_handler: NotificationHandler): - BaseActivity.__init__(self, logger, notification_handler) + BaseActivity.__init__( + self, logger, notification_handler, set_error_counter=True) @activity.defn(name="input_gate") async def input_gate(self, input_data: dict[str, Any]) -> tuple[str | None, int, str]: diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index af53a22..0d94d13 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -16,7 +16,8 @@ with workflow.unsafe.imports_passed_through(): class MLFlow(BaseActivity): def __init__(self, mlflow_host: str, mlflow_port: int, mlflow_username: str, mlflow_password: str, logger: Logger, notification_handler: NotificationHandler): - BaseActivity.__init__(self, logger, notification_handler) + BaseActivity.__init__( + self, logger, notification_handler, set_error_counter=True) self.mlflow_host = mlflow_host self.mlflow_port = mlflow_port self.mlflow_username = mlflow_username diff --git a/laborious/activities/opc.py b/laborious/activities/opc.py index 4ee3092..23e3ac4 100644 --- a/laborious/activities/opc.py +++ b/laborious/activities/opc.py @@ -51,7 +51,8 @@ class OPC(BaseActivity): attachment_content=error_data.get( 'attachment_content', None) ) - BaseActivity.__init__(self, logger, notification_handler) + BaseActivity.__init__( + self, logger, notification_handler, set_error_counter=True) def write_data(self, server_id: str, tag: str, data: Any, data_type: str, tag_type: str, metadata: dict[str, Any]) -> bool: