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.
This commit is contained in:
@@ -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')
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user