From 07891ac32ebfbf7b9ac7f94a8d70ad11ebb6bc0f Mon Sep 17 00:00:00 2001 From: Aethor Date: Tue, 19 Mar 2024 11:25:01 +0100 Subject: [PATCH] critical fix for score_muc --- pyproject.toml | 2 +- tibert/score.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index f804d07..708865d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "tibert" -version = "0.3.0" +version = "0.3.1" description = "BERT for Coreference Resolution" authors = ["Arthur Amalvy "] license = "GPL-3.0-only" diff --git a/tibert/score.py b/tibert/score.py index 510353a..4c47846 100644 --- a/tibert/score.py +++ b/tibert/score.py @@ -115,7 +115,7 @@ def score_muc( np.int = int # type: ignore np.bool = bool # type: ignore - precisions, recalls, f1s = [] + precisions, recalls, f1s = [], [], [] for pred, ref in zip(preds, refs): p, r, f1 = _neleval_precision_recall_f1(pred, ref, muc) precisions.append(p)