SIENTIAPDE-1172
Update GITHUB_BRANCH in values.yaml and enhance error notification handling in MLFlow - Changed GITHUB_BRANCH in values.yaml to reflect the new pipeline for alerts orchestration. - Added NotificationLevel.ERROR to error notifications in MLFlow for retraining and production model updates. - Updated tests to verify error notification levels for model management workflows.
This commit is contained in:
@@ -4,6 +4,7 @@ import numpy as np
|
||||
from pandas import DataFrame
|
||||
from pytest import fixture, mark
|
||||
from laborious.activities.mlflow import MLFlow
|
||||
from sientia_do.notifications.models import NotificationLevel
|
||||
|
||||
|
||||
@patch("laborious.activities.mlflow.MLFlowRepository")
|
||||
@@ -203,6 +204,7 @@ async def test_retrain_model_error(mlflow):
|
||||
notification_id='RETRAIN_MODEL_ERROR',
|
||||
message='Error retraining model test_model: Error retraining model',
|
||||
block='retrain_model',
|
||||
level=NotificationLevel.ERROR,
|
||||
attachment_content=ANY
|
||||
)
|
||||
else:
|
||||
@@ -266,6 +268,7 @@ async def test_update_production_model_error(mlflow):
|
||||
notification_id='UPDATE_PRODUCTION_MODEL_ERROR',
|
||||
message='Error updating production model test_model: Error updating production model',
|
||||
block='update_production_model',
|
||||
level=NotificationLevel.ERROR,
|
||||
attachment_content=ANY
|
||||
)
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user