Enable WandB logging and add API key

requirements.txt

@@ -9,3 +9,5 @@ accelerate
 pytest
 ruff
 mypy
+
+wandb

trainer/grpo_config.py

@@ -22,6 +22,6 @@ def build_config(run_name="codeforge-grpo", output_dir="checkpoints"):
         logging_steps=1, save_steps=100,
         save_total_limit=3,
         output_dir=output_dir,
-        # report_to="wandb",  # Disabled to avoid token requirements unless needed
+        report_to="wandb",
         run_name=run_name,
     )

trainer/train.py

@@ -1,6 +1,9 @@
 import sys
 import os
 
+os.environ["WANDB_API_KEY"] = "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"
+os.environ["WANDB_PROJECT"] = "codeforge-grpo"
+
 # Add the root directory to PYTHONPATH so codeforge module can be imported
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))