From 47885e743b1b25eb0202a1dd6681480452587501 Mon Sep 17 00:00:00 2001
From: Yury Kirpichev <ykirpichev@gmail.com>
Date: Fri, 2 Jan 2026 10:53:02 -0800
Subject: [PATCH] Fix WANDB_RUN parameter passing in runcpu.sh

- Add --run=$WANDB_RUN to base_train, mid_train, and chat_sft calls
- Ensures wandb logging works when WANDB_RUN environment variable is set
- Matches the behavior in speedrun.sh
---
 dev/runcpu.sh | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/dev/runcpu.sh b/dev/runcpu.sh
index ffacefa..73ef6ef 100755
--- a/dev/runcpu.sh
+++ b/dev/runcpu.sh
@@ -44,7 +44,8 @@ python -m scripts.base_train \
     --core_metric_every=50 \
     --core_metric_max_per_task=12 \
     --sample_every=50 \
-    --num_iterations=50
+    --num_iterations=50 \
+    --run=$WANDB_RUN
 python -m scripts.base_loss --device_batch_size=1 --split_tokens=4096
 python -m scripts.base_eval --max-per-task=16
 
@@ -55,7 +56,8 @@ python -m scripts.mid_train \
     --eval_every=50 \
     --eval_tokens=4096 \
     --total_batch_size=1024 \
-    --num_iterations=100
+    --num_iterations=100 \
+    --run=$WANDB_RUN
 # eval results will be terrible, this is just to execute the code paths.
 # note that we lower the execution memory limit to 1MB to avoid warnings on smaller systems
 python -m scripts.chat_eval --source=mid --max-new-tokens=128 --max-problems=20
@@ -66,7 +68,8 @@ python -m scripts.chat_sft \
     --target_examples_per_step=4 \
     --num_iterations=100 \
     --eval_steps=4 \
-    --eval_metrics_max_problems=16
+    --eval_metrics_max_problems=16 \
+    --run=$WANDB_RUN
 
 # Chat CLI
 # python -m scripts.chat_cli -p "Why is the sky blue?"