Update simuleval/evaluator/instance.py

Co-authored-by: Abinesh Ramakrishnan <3632454+ibanesh@users.noreply.github.com>
facebookresearch · Nov 29, 2023 · 261d9b7 · 261d9b7
1 parent 3da63af
commit 261d9b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simuleval/evaluator/instance.py b/simuleval/evaluator/instance.py
@@ -465,7 +465,7 @@ def __init__(self, info: str) -> None:
 
         self.index = self.info["index"]
         self.reference = self.info.get("reference", "")
-        self.latency_unit = "word"
+        self.latency_unit = latency_unit
         self.source_length = self.info.get("source_length")  # just for testing!
         self.finish_prediction = True
         self.metrics = {}