diff --git a/infra/k8s/argo/train-multi-seed-template.yaml b/infra/k8s/argo/train-multi-seed-template.yaml index 03ecb9f65..95a18c922 100644 --- a/infra/k8s/argo/train-multi-seed-template.yaml +++ b/infra/k8s/argo/train-multi-seed-template.yaml @@ -252,7 +252,7 @@ spec: echo "Resolved SHA: $SHA (short: $SHORT_SHA)" case "$MODEL" in - alpha-rl) BINARIES="alpha_rl_train precompute_features" ;; + alpha-rl) BINARIES="alpha_rl_train" ;; dqn|ppo) BINARIES="train_baseline_rl evaluate_baseline precompute_features" ;; *) BINARIES="train_baseline_supervised evaluate_supervised precompute_features" ;; esac @@ -287,7 +287,6 @@ spec: if [ "$MODEL" = "alpha-rl" ]; then cargo build --release -p ml-alpha --example alpha_rl_train - cargo build --release -p ml --features ml/cuda --example precompute_features else ML_EXAMPLE_ARGS="" for ex in $BINARIES; do