SIENTIAPDE-1231
Enhance Gates and MLFlowRepository with new functionalities and improvements - Added a new method `clean_tmp_files` in the Gates class to remove temporary files associated with model retraining. - Updated MLFlowRepository methods to improve experiment handling, including dynamic parameter logging and model retrieval. - Refactored model loading methods to streamline the process and enhance error handling. - Improved logging for model operations and added support for model parameter retrieval. - Adjusted minimal_retrain workflow to extend timeouts for activities and ensure proper model configuration handling.
This commit is contained in:
@@ -95,7 +95,7 @@ class MinimalRetrain():
|
||||
'model_config': model_config
|
||||
},
|
||||
retry_policy=retry_policy,
|
||||
start_to_close_timeout=timedelta(seconds=600)
|
||||
start_to_close_timeout=timedelta(hours=1)
|
||||
)
|
||||
|
||||
if experiment_response['success']:
|
||||
@@ -108,7 +108,7 @@ class MinimalRetrain():
|
||||
**experiment_response
|
||||
},
|
||||
retry_policy=retry_policy,
|
||||
start_to_close_timeout=timedelta(seconds=600)
|
||||
start_to_close_timeout=timedelta(seconds=60)
|
||||
)
|
||||
else:
|
||||
update_report = {}
|
||||
@@ -123,7 +123,7 @@ class MinimalRetrain():
|
||||
'update_report': update_report
|
||||
},
|
||||
retry_policy=retry_policy,
|
||||
start_to_close_timeout=timedelta(seconds=600)
|
||||
start_to_close_timeout=timedelta(seconds=60)
|
||||
)
|
||||
|
||||
await workflow.execute_activity_method(
|
||||
|
||||
Reference in New Issue
Block a user