From c42bf033e5a7835ea1664e23238d92ee1d13295a Mon Sep 17 00:00:00 2001 From: Max Date: Fri, 8 May 2026 14:46:24 -0500 Subject: [PATCH] training/fleet/manifest: accept knn + knn_semi in _ALLOWED_MODELS Validator's allowed-models frozenset was missing knn and knn_semi even though the manifest gained those jobs and the model registry registered the classes. Lambda bootstrap blocked at: TrainingManifestError: job 'knn-realistic': model 'knn' not in ['cnn', 'gbt', 'gru', 'lstm', 'mlp', 'transformer', 'transformer_ssl'] Now {gbt, knn, knn_semi, mlp, cnn, gru, lstm, transformer, transformer_ssl}. Co-Authored-By: Claude Opus 4.7 (1M context) --- training/fleet/manifest.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/training/fleet/manifest.py b/training/fleet/manifest.py index 24bafa8..7a37ee9 100644 --- a/training/fleet/manifest.py +++ b/training/fleet/manifest.py @@ -110,7 +110,8 @@ class TrainingManifest: # Allowed model names — keep in sync with training/models/REGISTRY _ALLOWED_MODELS = frozenset({ - "gbt", "mlp", "cnn", "gru", "lstm", "transformer", "transformer_ssl", + "gbt", "knn", "knn_semi", + "mlp", "cnn", "gru", "lstm", "transformer", "transformer_ssl", }) _ALLOWED_MODES = frozenset({"realistic", "oracle"}) _ALLOWED_RECIPES = frozenset({"host", "sample", "time"})