diff --git a/src/open_r1/utils/upload_details.py b/scripts/upload_details.py
similarity index 100%
rename from src/open_r1/utils/upload_details.py
rename to scripts/upload_details.py
diff --git a/slurm/evaluate.slurm b/slurm/evaluate.slurm
index c659c0b3..da106f6b 100644
--- a/slurm/evaluate.slurm
+++ b/slurm/evaluate.slurm
@@ -81,7 +81,7 @@ echo "Uploading details to Hugging Face Hub..."
 DETAILS_FILEPATHS=$(find $OUTPUT_DIR/details/ -type f \( -name "*.parquet" \))
 echo "DETAILS_FILEPATHS: $DETAILS_FILEPATHS"
 TIMESTAMP=$(date +"%Y-%m-%dT%H-%M-%S")
-python src/open_r1/utils/upload_details.py --data_files $DETAILS_FILEPATHS --hub_repo_id $DETAILS_REPO_ID --config_name $MODEL_REVISION.$TASK_NAME.$TIMESTAMP
+python scripts/upload_details.py --data_files $DETAILS_FILEPATHS --hub_repo_id $DETAILS_REPO_ID --config_name $MODEL_REVISION.$TASK_NAME.$TIMESTAMP
     
 echo "Cleaning up ..."
 rm -rf $OUTPUT_DIR
diff --git a/src/open_r1/utils/logging.py b/src/open_r1/utils/logging.py
deleted file mode 100644
index 764f30f8..00000000
--- a/src/open_r1/utils/logging.py
+++ /dev/null
@@ -1,9 +0,0 @@
-import os
-
-
-def init_wandb_training(training_args):
-    """
-    Helper function for setting up Weights & Biases logging tools.
-    """
-    os.environ["WANDB_ENTITY"] = training_args.wandb_entity
-    os.environ["WANDB_PROJECT"] = training_args.wandb_project
diff --git a/src/open_r1/utils/wandb_logging.py b/src/open_r1/utils/wandb_logging.py
new file mode 100644
index 00000000..13b55276
--- /dev/null
+++ b/src/open_r1/utils/wandb_logging.py
@@ -0,0 +1,11 @@
+import os
+
+
+def init_wandb_training(training_args):
+    """
+    Helper function for setting up Weights & Biases logging tools.
+    """
+    if training_args.wandb_entity is not None:
+        os.environ["WANDB_ENTITY"] = training_args.wandb_entity
+    if training_args.wandb_project is not None:
+        os.environ["WANDB_PROJECT"] = training_args.wandb_project