fix: update bert_score test assertion

IAAR-Shanghai · Aug 19, 2024 · 7606739 · 7606739
1 parent f6cb730
commit 7606739
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_metrics.py b/tests/test_metrics.py
@@ -26,7 +26,7 @@ def test_keyword_precision(self):
     def test_bert_score(self):
         continuation = "这是一个测试"
         reference = "这是一个测试"
-        self.assertEqual(bert_score(continuation, reference), 1.0)
+        self.assertAlmostEquals(bert_score(continuation, reference), 1.0, 6)
 
 
 if __name__ == "__main__":