update

Author: nrimsky

behaviors.py

@@ -181,11 +181,11 @@ def get_finetuned_model_path(
 
 
 def get_finetuned_model_results_path(
-    behavior: str, pos_or_neg: Optional[Literal["pos", "neg"]], layer=None
+    behavior: str, pos_or_neg: Optional[Literal["pos", "neg"]], eval_type: str, layer=None
 ) -> str:
     if layer is None:
         layer = "all"
     return os.path.join(
         RESULTS_PATH,
-        f"{behavior}_{pos_or_neg}_finetune_{layer}_results.json",
+        f"{behavior}_{pos_or_neg}_finetune_{layer}_{eval_type}_results.json",
     )

eval_finetune_llama.py

@@ -118,7 +118,7 @@ def test_finetune(
 ):
     save_results_dir = get_results_dir(behavior)
     finetuned_model_path = get_finetuned_model_path(behavior, pos_or_neg, layer)
-    save_filename = get_finetuned_model_results_path(behavior, pos_or_neg, layer)
+    save_filename = get_finetuned_model_results_path(behavior, pos_or_neg, eval_type=test_type, layer=layer)
     if not os.path.exists(save_results_dir):
         os.makedirs(save_results_dir)
     process_methods = {