From 729e8b5fae604fcbd537bc226dc81f597b9e4a3f Mon Sep 17 00:00:00 2001
From: jgrusewski <jgrusewski@localhost>
Date: Fri, 3 Apr 2026 23:00:14 +0200
Subject: [PATCH] feat: argo-train.sh --baseline flag (skips hyperopt) +
 --cache-dir

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 scripts/argo-train.sh | 29 +++++++++++++++++++++--------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/scripts/argo-train.sh b/scripts/argo-train.sh
index 8cbb68e71..ba6aae075 100755
--- a/scripts/argo-train.sh
+++ b/scripts/argo-train.sh
@@ -22,6 +22,8 @@ SYMBOL=""
 WEBHOOK=false
 WEBHOOK_BASE="http://workflow-trigger-eventsource-svc.foxhunt:12001"
 WATCH=false
+BASELINE=false
+CACHE_DIR=""
 
 usage() {
   cat <<EOF
@@ -37,6 +39,8 @@ Options:
   --epochs <n>           Training epochs (default: model-specific)
   --gpu-pool <pool>      GPU node pool (default: ci-training-h100)
   --symbol <sym>         Trading symbol (default: ES.FUT)
+  --baseline             Skip hyperopt (set trials=0), train with defaults
+  --cache-dir <path>     Override feature cache directory
   --webhook              Trigger via webhook instead of argo CLI
   --watch                Follow workflow logs after submission
   -h, --help             Show this help
@@ -61,6 +65,8 @@ while [[ $# -gt 0 ]]; do
     --epochs)    EPOCHS="$2"; shift 2 ;;
     --gpu-pool)  GPU_POOL="$2"; shift 2 ;;
     --symbol)    SYMBOL="$2"; shift 2 ;;
+    --baseline)  BASELINE=true; shift ;;
+    --cache-dir) CACHE_DIR="$2"; shift 2 ;;
     --webhook)   WEBHOOK=true; shift ;;
     --watch)     WATCH=true; shift ;;
     -h|--help)   usage ;;
@@ -99,10 +105,15 @@ if [[ "$MODEL" != "dqn" && "$MODEL" != "ppo" ]]; then
   CMD="$CMD -p model=$MODEL"
 fi
 
-[[ -n "$TRIALS" ]]   && CMD="$CMD -p hyperopt-trials=$TRIALS"
-[[ -n "$EPOCHS" ]]   && CMD="$CMD -p train-epochs=$EPOCHS"
-[[ -n "$GPU_POOL" ]] && CMD="$CMD -p gpu-pool=$GPU_POOL"
-[[ -n "$SYMBOL" ]]   && CMD="$CMD -p symbol=$SYMBOL"
+[[ -n "$TRIALS" ]]    && CMD="$CMD -p hyperopt-trials=$TRIALS"
+[[ -n "$EPOCHS" ]]    && CMD="$CMD -p train-epochs=$EPOCHS"
+[[ -n "$GPU_POOL" ]]  && CMD="$CMD -p gpu-pool=$GPU_POOL"
+[[ -n "$SYMBOL" ]]    && CMD="$CMD -p symbol=$SYMBOL"
+[[ -n "$CACHE_DIR" ]] && CMD="$CMD -p feature-cache-dir=$CACHE_DIR"
+
+if $BASELINE; then
+  CMD="$CMD -p hyperopt-trials=0"
+fi
 
 if $WATCH; then
   CMD="$CMD --watch"
@@ -110,9 +121,11 @@ fi
 
 echo "Submitting $MODEL training workflow ($TEMPLATE)..."
 echo "  binary-tag: $TAG"
-[[ -n "$TRIALS" ]]   && echo "  trials:     $TRIALS"
-[[ -n "$EPOCHS" ]]   && echo "  epochs:     $EPOCHS"
-[[ -n "$GPU_POOL" ]] && echo "  gpu-pool:   $GPU_POOL"
-[[ -n "$SYMBOL" ]]   && echo "  symbol:     $SYMBOL"
+$BASELINE                && echo "  mode:       baseline (hyperopt skipped)"
+[[ -n "$TRIALS" ]]       && echo "  trials:     $TRIALS"
+[[ -n "$EPOCHS" ]]       && echo "  epochs:     $EPOCHS"
+[[ -n "$GPU_POOL" ]]     && echo "  gpu-pool:   $GPU_POOL"
+[[ -n "$SYMBOL" ]]       && echo "  symbol:     $SYMBOL"
+[[ -n "$CACHE_DIR" ]]    && echo "  cache-dir:  $CACHE_DIR"
 echo ""
 eval "$CMD"