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

fix(worker): use VALID_MODELS from contracts in validation
Build and Push Docker Images / build-and-push (push) Waiting to run Details

This commit is contained in:
fchinembiri 2026-05-01 12:56:34 +02:00
parent 9b9bd2da5e
commit 084170ddc0
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/worker/worker.py
View File

@ -171,9 +171,9 @@ def parse_and_validate_payload(payload: dict) -> tuple[dict, List[str]]:
# Validate model # Validate model
if "model" in payload: if "model" in payload:
valid_models = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "CatBoost_V2"] from contracts import VALID_MODELS
if payload["model"] not in valid_models: if payload["model"] not in VALID_MODELS:
errors.append(f"Invalid model: {payload['model']}. Must be one of {valid_models}") errors.append(f"Invalid model: {payload['model']}. Must be one of {VALID_MODELS}")
# Validate kernel # Validate kernel
if "smoothing_kernel" in payload: if "smoothing_kernel" in payload: