more paths fixing later

src/gen/gen_judgment.py

@@ -6,6 +6,7 @@ import os
 import re
 
 import huggingface_hub
+from envs import HF_HOME
 from tqdm import tqdm
 from utils import (
     chat_completion_anthropic,
@@ -126,8 +127,8 @@ def judgment(**args):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--setting-file", type=str, default="./config/judge_config.yaml")
-    parser.add_argument("--endpoint-file", type=str, default="./config/api_config.yaml")
+    parser.add_argument("--setting-file", type=str, default=f"{HF_HOME}/src/gen/config/judge_config.yaml")
+    parser.add_argument("--endpoint-file", type=str, default="{HF_HOME}/src/gen/config/api_config.yaml")
     args = parser.parse_args()
     print(args)
 
@@ -142,10 +143,10 @@ if __name__ == "__main__":
     if configs["regex_pattern"]:
         pattern = re.compile(configs["regex_pattern"])
 
-    question_file = os.path.join("./data", configs["bench_name"], "question.jsonl")
-    external_dir = os.path.join("./data", configs["bench_name"], "model_answers/external")
-    internal_dir = os.path.join("./data", configs["bench_name"], "model_answers/internal")
-    ref_answer_dir = os.path.join("data", configs["bench_name"], "reference_answer")
+    question_file = os.path.join(f"{HF_HOME}/data", configs["bench_name"], "question.jsonl")
+    internal_dir = os.path.join(f"{HF_HOME}/data", configs["bench_name"], "model_answers/internal")
+    external_dir = os.path.join(f"{HF_HOME}/data", configs["bench_name"], "model_answers/external")
+    ref_answer_dir = os.path.join(f"{HF_HOME}/data", configs["bench_name"], "reference_answer")
 
     questions = load_questions(question_file)
     model_answers_external = load_model_answers(external_dir)
@@ -157,7 +158,7 @@ if __name__ == "__main__":
     # if user choose a set of models, only judge those models
     models = [
         model.split("/")[-1].split(".")[0]
-        for model in glob.glob("./data/arena-hard-v0.1/model_answers/external/*.jsonl")
+        for model in glob.glob(f"{HF_HOME}/data/arena-hard-v0.1/model_answers/external/*.jsonl")
     ]
 
     ref_answers = None
@@ -166,7 +167,7 @@ if __name__ == "__main__":
         ref_answers = [ref_answers[model] for model in configs["ref_model"]]
 
     output_files = {}
-    output_dir = f"data/{configs['bench_name']}/model_judgment/{configs['judge_model']}"
+    output_dir = f"{HF_HOME}/data/{configs['bench_name']}/model_judgment/{configs['judge_model']}"
     for model in models:
         output_files[model] = os.path.join(
             output_dir,