Fix bug in writing diagnostic preds
@@ -406,7 +406,7 @@ def _write_diagnostics_preds(
with open(preds_file, "w", encoding="utf-8") as preds_fh:
for row_idx, row in preds_df.iterrows():
if strict_glue_format:
- out_d = {"idx": row["idx"], "label": pred_map[row["labels"]]}
+ out_d = {"idx": row["idx"], "label": pred_map[row["preds"]]}
else:
out_d = row.to_dict()
preds_fh.write("{0}\n".format(json.dumps(out_d)))
