From e1d1e3d43d1404e5de22fde22d77a4673d9cf263 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Mon, 23 Mar 2026 08:06:01 -0300 Subject: [PATCH] SIENTIAPDE-1712 Refactor imports in gates.py and mlflow.py for improved organization - Removed unnecessary import statement in mlflow.py and re-added it in a more appropriate location. - Cleaned up the workflow metadata assignment in gates.py for better readability. --- laborious/activities/gates.py | 2 +- laborious/activities/mlflow.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py index 7acafd3..0b747f9 100644 --- a/laborious/activities/gates.py +++ b/laborious/activities/gates.py @@ -479,7 +479,7 @@ class Gates(MinioManager): minio_repo=self.minio_repository, model_name=input_data['model_name'], operation='transform', - workflow_metadata=metadata, + workflow_metadata=metadata ) @activity.defn(name='format_prediction') diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 27d1b92..e6c75e1 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -1,6 +1,6 @@ from temporalio import activity, workflow -from laborious.utils.repository.minio_manager import MinioManager + with workflow.unsafe.imports_passed_through(): import traceback @@ -20,7 +20,7 @@ with workflow.unsafe.imports_passed_through(): now, ) from sientia_do.utils.formatters import create_sample_dict - + from laborious.utils.repository.minio_manager import MinioManager from laborious.utils.models.minio_dataframe_payload import MinioDataFramePayload from laborious.utils.repository.model_repository import MLFlowRepository