From 0b06794fde09bedfb75ee85504148cf3a4707e21 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 13 三月 2023 20:14:41 +0800
Subject: [PATCH] bug fix

---
 funasr/utils/timestamp_tools.py |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/funasr/utils/timestamp_tools.py b/funasr/utils/timestamp_tools.py
index 7ba3e0b..2bccd50 100644
--- a/funasr/utils/timestamp_tools.py
+++ b/funasr/utils/timestamp_tools.py
@@ -170,8 +170,9 @@
         return uttid_list, ts_dict
 
     def _shift(self, filtered_timestamp_list1, filtered_timestamp_list2):
+        shift_time = 0
         for fts1, fts2 in zip(filtered_timestamp_list1, filtered_timestamp_list2):
-            shift_time = abs(fts1[0] - fts2[0]) + abs(fts1[1] - fts2[1])
+            shift_time += abs(fts1[0] - fts2[0]) + abs(fts1[1] - fts2[1])
         num_tokens = len(filtered_timestamp_list1)
         return shift_time, num_tokens
 

--
Gitblit v1.9.1