Fixed bug in wers module

analyticsinmotion · Nov 9, 2023 · 4808e39 · 4808e39
1 parent 629b78c
commit 4808e39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/werpy/wers.py b/werpy/wers.py
@@ -51,5 +51,5 @@ def wers(reference, hypothesis):
         transform_word_error_rate_breakdown = np.transpose(word_error_rate_breakdown.tolist())
         wers_result = transform_word_error_rate_breakdown[0].tolist()
     else:
-        wers_result = word_error_rate_breakdown[0].tolist()
+        wers_result = word_error_rate_breakdown[0]
     return wers_result