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,
|
minio_repo=self.minio_repository,
|
||||||
model_name=input_data['model_name'],
|
model_name=input_data['model_name'],
|
||||||
operation='transform',
|
operation='transform',
|
||||||
workflow_metadata=metadata,
|
workflow_metadata=metadata
|
||||||
)
|
)
|
||||||
|
|
||||||
@activity.defn(name='format_prediction')
|
@activity.defn(name='format_prediction')
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
from temporalio import activity, workflow
|
from temporalio import activity, workflow
|
||||||
|
|
||||||
from laborious.utils.repository.minio_manager import MinioManager
|
|
||||||
|
|
||||||
with workflow.unsafe.imports_passed_through():
|
with workflow.unsafe.imports_passed_through():
|
||||||
import traceback
|
import traceback
|
||||||
@@ -20,7 +20,7 @@ with workflow.unsafe.imports_passed_through():
|
|||||||
now,
|
now,
|
||||||
)
|
)
|
||||||
from sientia_do.utils.formatters import create_sample_dict
|
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.models.minio_dataframe_payload import MinioDataFramePayload
|
||||||
from laborious.utils.repository.model_repository import MLFlowRepository
|
from laborious.utils.repository.model_repository import MLFlowRepository
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user