daiteng01 commited on
Commit
785edf8
·
1 Parent(s): 8559e9f
Files changed (1) hide show
  1. src/tools/plots.py +1 -1
src/tools/plots.py CHANGED
@@ -39,7 +39,7 @@ def create_scores_df(raw_data: list[EvalResult]) -> pd.DataFrame:
39
  if task.benchmark == "Average":
40
  current_score = np.mean(list(row["results"].values()))
41
  else:
42
- if row["results"]:
43
  current_score = row["results"][task.benchmark]
44
  else:
45
  current_score = 0
 
39
  if task.benchmark == "Average":
40
  current_score = np.mean(list(row["results"].values()))
41
  else:
42
+ if row["results"] and task.benchmark in row["results"]:
43
  current_score = row["results"][task.benchmark]
44
  else:
45
  current_score = 0