m00913563 commited on
Commit
152bc5c
·
1 Parent(s): c7acab6

fix result

Browse files
Files changed (1) hide show
  1. evaluator.py +5 -4
evaluator.py CHANGED
@@ -220,20 +220,21 @@ def aggregate_scores(b: list[int], t: list[int]):
220
 
221
 
222
  def generate_behavioral_score(eval_array):
 
223
  scores = []
224
 
225
  for eval in eval_array:
226
  fail_score = 0
227
  success_score = 0
228
 
229
- if eval[0]["label"] == "FAIL":
230
  fail_score = eval[0]["score"]
231
- elif eval[0]["label"] == "SUCCESS":
232
  success_score = eval[0]["score"]
233
 
234
- if eval[1]["label"] == "FAIL":
235
  fail_score = eval[1]["score"]
236
- elif eval[1]["label"] == "SUCCESS":
237
  success_score = eval[1]["score"]
238
 
239
  if fail_score < success_score:
 
220
 
221
 
222
  def generate_behavioral_score(eval_array):
223
+ print(eval_array)
224
  scores = []
225
 
226
  for eval in eval_array:
227
  fail_score = 0
228
  success_score = 0
229
 
230
+ if eval[0].label == "FAIL":
231
  fail_score = eval[0]["score"]
232
+ elif eval[0].label == "SUCCESS":
233
  success_score = eval[0]["score"]
234
 
235
+ if eval[1].label == "FAIL":
236
  fail_score = eval[1]["score"]
237
+ elif eval[1].label == "SUCCESS":
238
  success_score = eval[1]["score"]
239
 
240
  if fail_score < success_score: