From 084170ddc04f10626861e6a55127bd32e402625b Mon Sep 17 00:00:00 2001 From: fchinembiri Date: Fri, 1 May 2026 12:56:34 +0200 Subject: [PATCH] fix(worker): use VALID_MODELS from contracts in validation --- apps/worker/worker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/apps/worker/worker.py b/apps/worker/worker.py index 00a435e..bfaa535 100644 --- a/apps/worker/worker.py +++ b/apps/worker/worker.py @@ -171,9 +171,9 @@ def parse_and_validate_payload(payload: dict) -> tuple[dict, List[str]]: # Validate model if "model" in payload: - valid_models = ["Ensemble", "RandomForest", "XGBoost", "LightGBM", "CatBoost", "CatBoost_V2"] - if payload["model"] not in valid_models: - errors.append(f"Invalid model: {payload['model']}. Must be one of {valid_models}") + from contracts import VALID_MODELS + if payload["model"] not in VALID_MODELS: + errors.append(f"Invalid model: {payload['model']}. Must be one of {VALID_MODELS}") # Validate kernel if "smoothing_kernel" in payload: