fchinembiri
/
geocrop-platform.
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/worker/contracts.py
View File

@ -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]