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) <noreply@anthropic.com>
This commit is contained in:
Max 2026-05-08 14:46:24 -05:00
parent 4bf241f6ec
commit c42bf033e5

View file

@ -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"})