minor

dtch1997 · May 6, 2024 · 7ea4433 · 7ea4433
1 parent 78f9bc4
commit 7ea4433
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/repepo/experiments/persona_prompts.py b/repepo/experiments/persona_prompts.py
@@ -491,6 +491,12 @@
 
 
 PERSONA_PROMPTS: dict[str, tuple[str, str]] = {
+    # advanced ai risk
+    **ADVANCED_AI_RISK,
+    # CAA
+    **CAA_SYCOPHANCY_PROMPTS,
+    **CAA_TRUTHFULQA_PROMPTS,
+    # persona
     # topic: politics
     **POLITICS_PROMPTS,
     # topic: religion
@@ -537,10 +543,6 @@
     **SELF_PRESERVATION_PROMPTS,
     # misc ai risk
     **MISC_AI_RISK_PROMPTS,
-    # advanced ai risk
-    **ADVANCED_AI_RISK,
-    **CAA_SYCOPHANCY_PROMPTS,
-    **CAA_TRUTHFULQA_PROMPTS
 }
 
 CATEGORIZED_PERSONA_PROMPTS = {

diff --git a/scripts/persona_generalization.qsub.sh b/scripts/persona_generalization.qsub.sh
@@ -9,7 +9,7 @@
 #$ -t 1-116
 
 # Add locally installed executables to PATH
-source /home/$USER/.bash_profile
+source $HOME/.bash_profile
 
 # Start the job
 # NOTE: Assume you have imported cluster utils
@@ -18,14 +18,14 @@ send_slack_notification "Job $JOB_NAME:$JOB_ID:$SGE_TASK_ID started"
 
 # NOTE: Assume you have already cloned the repository and checked out correct branch
 # Navigate to the project directory
-cd /home/$USER/Scratch/repepo
+cd $HOME/Scratch/repepo
 
 # NOTE: Assume you have already installed pdm
 pdm install
 
 # Run the script
 pdm run python repepo/experiments/persona_generalization.py \
-    --output_dir repepo/experiments/persona_generalization \
+    --output_dir experiments/persona_generalization \
     --sge_task_id $SGE_TASK_ID
 
 # End the job