Always return a Python int as distance from _greedy_correct_assignment

fgnt · Aug 23, 2024 · c00969d · c00969d
1 parent cd1a36f
commit c00969d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/meeteval/wer/matching/greedy_combination_matching.py b/meeteval/wer/matching/greedy_combination_matching.py
@@ -151,7 +151,7 @@ def compute_backward_columns(r, h):
         # We know that the cost cannot increase, so check that we do correct
         assert sum(old_stream_costs) >= sum(stream_costs), (old_stream_costs, stream_costs)
 
-    return assignment, np.sum(stream_costs)
+    return assignment, int(np.sum(stream_costs))
 
 
 def initialize_assignment(
@@ -239,4 +239,4 @@ def greedy_combination_matching(
             functools.partial(cy_forward_col, cost_substitution=int(d))
         )
 
-    return int(distance), assignment
+    return distance, assignment