hf jobs: propagate TEST_ONLY env through launcher + remote runner

training/hf_remote_run.sh

@@ -17,6 +17,7 @@ set -euo pipefail
 
 NB_EXEC_TIMEOUT="${NB_EXEC_TIMEOUT:-14400}"
 export SMOKE_MODE="${SMOKE_MODE:-0}"
+export TEST_ONLY="${TEST_ONLY:-0}"
 export HF_HUB_ENABLE_HF_TRANSFER="${HF_HUB_ENABLE_HF_TRANSFER:-1}"
 export TOKENIZERS_PARALLELISM="${TOKENIZERS_PARALLELISM:-false}"
 export DEBIAN_FRONTEND=noninteractive
@@ -24,6 +25,7 @@ export DEBIAN_FRONTEND=noninteractive
 echo "===== hf_remote_run.sh starting at $(date -u +%FT%TZ) ====="
 echo "SPACE_REPO=${SPACE_REPO}"
 echo "SMOKE_MODE=${SMOKE_MODE}"
+echo "TEST_ONLY=${TEST_ONLY}"
 echo "NB_EXEC_TIMEOUT=${NB_EXEC_TIMEOUT}"
 
 nvidia-smi || true

training/hf_run_space_train_job.sh

@@ -16,6 +16,7 @@ TIMEOUT="${HF_JOB_TIMEOUT:-8h}"
 SPACE_REPO="${HF_SPACE_REPO_ID:-ycwhencpp/final-iteration}"
 NB_EXEC_TIMEOUT="${NB_EXEC_TIMEOUT:-14400}"
 SMOKE_MODE="${SMOKE_MODE:-0}"
+TEST_ONLY="${TEST_ONLY:-0}"
 
 if ! hf auth whoami &>/dev/null; then
   echo "Run: hf auth login" >&2
@@ -43,4 +44,5 @@ exec hf jobs run \
   --env "SPACE_REPO=$SPACE_REPO" \
   --env "NB_EXEC_TIMEOUT=$NB_EXEC_TIMEOUT" \
   --env "SMOKE_MODE=$SMOKE_MODE" \
+  --env "TEST_ONLY=$TEST_ONLY" \
   -- "$IMAGE" bash -c "$BOOTSTRAP"