diff --git a/orchestrator/activities/formatters.py b/orchestrator/activities/formatters.py index 636d8a9..4f741c8 100644 --- a/orchestrator/activities/formatters.py +++ b/orchestrator/activities/formatters.py @@ -58,10 +58,18 @@ schedule_types: dict[str, ScheduleType] = { 'namespace': 'laborious', 'function': predictions_batch, }, + 'xgboost_predictions_batch': { + 'namespace': 'laborious', + 'function': predictions_batch, + }, 'minimal_retrain': { 'namespace': 'laborious', 'function': minimal_retrain, }, + 'xgboost_minimal_retrain': { + 'namespace': 'laborious', + 'function': minimal_retrain, + }, 'drift': { 'namespace': 'laborious', 'function': drift,