SIENTIAPDE-1231

Update model retraining and logging enhancements

- Changed the GITHUB_BRANCH value in values.yaml to 'main' for consistency.
- Refactored MLFlow class to improve timestamp handling and error messaging during model retraining.
- Enhanced MLFlowRepository methods to include metadata logging and improved model version retrieval.
- Updated minimal_retrain workflow to support extended timeout for activities and include model configuration in input data.
This commit is contained in:
vitor-aignosi
2025-10-02 16:36:18 -03:00
parent c6f004d20d
commit 9b71ad7556
5 changed files with 204 additions and 150 deletions

View File

@@ -142,7 +142,8 @@ async def main():
activities.load_custom_query,
activities.retrain_model,
activities.update_production_model,
activities.export_data_to_postgres
activities.format_retrain_report,
activities.export_data_to_postgres,
],
max_concurrent_workflow_tasks=50,
max_concurrent_activities=50,