fix uv env path + details (#188)

* fix uv env path + details

* Update slurm/grpo.slurm

---------

Co-authored-by: lewtun <[email protected]>

Author: edbeeching

slurm/eval_callback.slurm

@@ -8,8 +8,7 @@
 
 set -x -e
 source ~/.bashrc
-conda activate openr1
-
+source openr1/bin/activate
 TASK_NAME=$1
 TASKS=$2
 MODEL_ID=$3
@@ -31,7 +30,7 @@ fi
 
 LM_EVAL_REPO_ID="open-r1/open-r1-eval-leaderboard"
 MODEL_NAME=$(echo $MODEL_ID | sed 's/\//_/g') # replaces / with _
-DETAILS_REPO_ID="open-r1//details-$MODEL_NAME"
+DETAILS_REPO_ID="open-r1/details-$MODEL_NAME"
 OUTPUT_DIR="eval_results/$MODEL_ID/$MODEL_REVISION/$TASK_NAME"
 # We need this flag since we run this script from training jobs that use DeepSpeed and the env vars get progated which causes errors during evaluation
 ACCELERATE_USE_DEEPSPEED=false

slurm/evaluate.slurm

@@ -14,7 +14,7 @@
 set -x -e
 
 source ~/.bashrc
-conda activate openr1
+source openr1/bin/activate
 module load cuda/12.1
 echo "START TIME: $(date)"
 echo "PYTHON ENV: $(which python)"

slurm/generate.slurm

@@ -129,7 +129,7 @@ export LD_LIBRARY_PATH=.venv/lib/python3.11/site-packages/nvidia/nvjitlink/lib
 echo "SLURM_JOB_ID: $SLURM_JOB_ID"
 echo "SLURM_JOB_NODELIST: $SLURM_JOB_NODELIST"
 
-source .venv/bin/activate
+source openr1/bin/activate
 
 # Getting the node names
 nodes=$(scontrol show hostnames "$SLURM_JOB_NODELIST")

slurm/grpo.slurm

@@ -11,7 +11,7 @@
 set -x -e
 
 source ~/.bashrc
-conda activate openr1
+source openr1/bin/activate
 echo "START TIME: $(date)"
 echo "PYTHON ENV: $(which python)"
 

slurm/sft.slurm

@@ -11,7 +11,7 @@
 set -x -e
 
 source ~/.bashrc
-conda activate openr1
+source openr1/bin/activate
 echo "START TIME: $(date)"
 echo "PYTHON ENV: $(which python)"
 

src/open_r1/utils/upload_details.py

@@ -39,7 +39,7 @@ class ScriptArguments:
 
 def main():
     parser = HfArgumentParser(ScriptArguments)
-    args = parser.parse()
+    args = parser.parse_args_into_dataclasses()[0]
 
     if all(file.endswith(".json") for file in args.data_files):
         ds = load_dataset("json", data_files=args.data_files)