replace depricated methods of jiwer

huggingface · Feb 1, 2025 · b84e7e1 · b84e7e1
1 parent 5aa3982
commit b84e7e1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/metrics/cer/cer.py b/metrics/cer/cer.py
@@ -137,23 +137,23 @@ def _info(self):
 
     def _compute(self, predictions, references, concatenate_texts=False):
         if concatenate_texts:
-            return jiwer.compute_measures(
+            return jiwer.process_words(
                 references,
                 predictions,
-                truth_transform=cer_transform,
+                reference_transform=cer_transform,
                 hypothesis_transform=cer_transform,
-            )["wer"]
+            ).wer
 
         incorrect = 0
         total = 0
         for prediction, reference in zip(predictions, references):
-            measures = jiwer.compute_measures(
+            out = jiwer.process_words(
                 reference,
                 prediction,
-                truth_transform=cer_transform,
+                reference_transform=cer_transform,
                 hypothesis_transform=cer_transform,
             )
-            incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"]
-            total += measures["substitutions"] + measures["deletions"] + measures["hits"]
+            incorrect += out.substitutions + out.deletions + out.insertions
+            total += out.substitutions + out.deletions + out.hits
 
         return incorrect / total
diff --git a/metrics/wer/wer.py b/metrics/wer/wer.py
@@ -14,7 +14,7 @@
 """ Word Error Ratio (WER) metric. """
 
 import datasets
-from jiwer import compute_measures
+from jiwer import process_words
 
 import evaluate
 
@@ -95,12 +95,12 @@ def _info(self):
 
     def _compute(self, predictions=None, references=None, concatenate_texts=False):
         if concatenate_texts:
-            return compute_measures(references, predictions)["wer"]
+            return process_words(references, predictions).wer
         else:
             incorrect = 0
             total = 0
             for prediction, reference in zip(predictions, references):
-                measures = compute_measures(reference, prediction)
-                incorrect += measures["substitutions"] + measures["deletions"] + measures["insertions"]
-                total += measures["substitutions"] + measures["deletions"] + measures["hits"]
+                out = process_words(reference, prediction)
+                incorrect += out.substitutions + out.deletions + out.insertions
+                total += out.substitutions + out.deletions + out.hits
             return incorrect / total