Просмотр исходного кода

fixbug path_name_type_list can [[any,str,str],[any,str,str]]

志浩 3 лет назад
Родитель
Сommit
0ac06c029e
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      funasr/datasets/iterable_dataset.py

+ 2 - 1
funasr/datasets/iterable_dataset.py

@@ -8,6 +8,7 @@ from typing import Dict
 from typing import Iterator
 from typing import Tuple
 from typing import Union
+from typing import List
 
 import kaldiio
 import numpy as np
@@ -127,7 +128,7 @@ class IterableESPnetDataset(IterableDataset):
         non_iterable_list = []
         self.path_name_type_list = []
 
-        if not isinstance(path_name_type_list[0], Tuple):
+        if not isinstance(path_name_type_list[0], (Tuple, List)):
             path = path_name_type_list[0]
             name = path_name_type_list[1]
             _type = path_name_type_list[2]