From f97bd14a7f19cbf2181c0546685bcd3a6bc5b0e9 Mon Sep 17 00:00:00 2001 From: fchinembiri Date: Fri, 1 May 2026 09:45:12 +0200 Subject: [PATCH] feat(worker): add Hybrid_SpatioTemporal to valid models --- apps/worker/contracts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/worker/contracts.py b/apps/worker/contracts.py index 470b667..ce430be 100644 --- a/apps/worker/contracts.py +++ b/apps/worker/contracts.py @@ -26,7 +26,7 @@ STAGES = [ ] # Acceptable model names -VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost"] +VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "Hybrid_SpatioTemporal"] # Valid smoothing kernel sizes VALID_KERNEL_SIZES = [3, 5, 7]