diff --git a/orchestrator/utils/orchestrator_functions.py b/orchestrator/utils/orchestrator_functions.py index f1d6510..6fb3810 100644 --- a/orchestrator/utils/orchestrator_functions.py +++ b/orchestrator/utils/orchestrator_functions.py @@ -13,6 +13,17 @@ def common_config(config: dict[str, Any]): } +def minimal_retrain(config: dict[str, Any]): + return { + **common_config(config), + "workflow_type": "retrain", + "schedule_name": config['schedule_name'], + "query": config['query'], + "schema": "sientia_data", + "table_name": "log_retrain", + } + + def scouter(config: dict[str, Any]): filters = {} for f in config.get('filters', []):