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]