feat(worker): add Hybrid_SpatioTemporal to valid models
Build and Push Docker Images / build-and-push (push) Successful in 15m45s Details

This commit is contained in:
fchinembiri 2026-05-01 09:45:12 +02:00
parent cd9f6f210d
commit f97bd14a7f
1 changed files with 1 additions and 1 deletions

View File

@ -26,7 +26,7 @@ STAGES = [
] ]
# Acceptable model names # Acceptable model names
VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost"] VALID_MODELS = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "Hybrid_SpatioTemporal"]
# Valid smoothing kernel sizes # Valid smoothing kernel sizes
VALID_KERNEL_SIZES = [3, 5, 7] VALID_KERNEL_SIZES = [3, 5, 7]