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: