From ba73bb7e9dacf63a1a37c54b855475e2aa0445a3 Mon Sep 17 00:00:00 2001
From: vitor-aignosi <vitor.santos@aignosi.com.br>
Date: Thu, 15 Jan 2026 11:08:48 -0300
Subject: [PATCH] SIENTIAPDE-1478  Add xgboost functions to schedule_types in
 formatters.py for enhanced orchestration capabilities

---
 orchestrator/activities/formatters.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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,