IliaLarchenko commited on
Commit
bb0f942
1 Parent(s): 1e66988

Small fixes for analysis

Browse files
Files changed (2) hide show
  1. tests/analysis.py +1 -1
  2. tests/grader.py +1 -1
tests/analysis.py CHANGED
@@ -125,7 +125,7 @@ def run_evaluation(
125
 
126
  exp_name = f"{exp_name}_{pd.Timestamp.now().strftime('%Y-%m-%d_%H-%M-%S')}"
127
  os.makedirs(f"records/{exp_name}", exist_ok=True)
128
- tasks = [(type_, i, llm_config) for type_ in interview_types for i in range(num_attempts) for llm_config in llm_configs]
129
  complete_f = partial(complete_and_grade, exp_name=exp_name, grader_models=grader_models, candidate_model=candidate_model)
130
 
131
  with concurrent.futures.ThreadPoolExecutor(max_workers=num_workers) as executor:
 
125
 
126
  exp_name = f"{exp_name}_{pd.Timestamp.now().strftime('%Y-%m-%d_%H-%M-%S')}"
127
  os.makedirs(f"records/{exp_name}", exist_ok=True)
128
+ tasks = [(type_, i, llm_config) for i in range(num_attempts) for type_ in interview_types for llm_config in llm_configs]
129
  complete_f = partial(complete_and_grade, exp_name=exp_name, grader_models=grader_models, candidate_model=candidate_model)
130
 
131
  with concurrent.futures.ThreadPoolExecutor(max_workers=num_workers) as executor:
tests/grader.py CHANGED
@@ -28,7 +28,7 @@ def grade(json_file_path, model="gpt-4-turbo", suffix=""):
28
  {"role": "user", "content": f"Please evaluate the interviewer based on the following data: \n {'\n'.join(interview_summary_list)}"},
29
  ]
30
 
31
- response = client.chat.completions.create(model=model, messages=messages, temperature=1, response_format={"type": "json_object"})
32
  feedback = json.loads(response.choices[0].message.content)
33
 
34
  feedback["file_name"] = json_file_path
 
28
  {"role": "user", "content": f"Please evaluate the interviewer based on the following data: \n {'\n'.join(interview_summary_list)}"},
29
  ]
30
 
31
+ response = client.chat.completions.create(model=model, messages=messages, temperature=0, response_format={"type": "json_object"})
32
  feedback = json.loads(response.choices[0].message.content)
33
 
34
  feedback["file_name"] = json_file_path