浏览代码

do not skip files

Loann Peurey 5 月之前
父节点
当前提交
c16ead672a
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      compute_annotations/compute_derived_annotations.py

+ 2 - 2
compute_annotations/compute_derived_annotations.py

@@ -66,6 +66,7 @@ def get_available_segments(project_path, set_name, base_sets, raw_recording_avai
 
     # Get already existing conversation annotations and only compute annotations for the files that do not already
     # have conversational annotations
+    set_name = set_name.lower()
     if set_name in set(am.annotations['set']):
         target_set_anns = am.annotations[am.annotations['set'] == set_name]
         available_base_sets_anns = available_base_sets_anns[
@@ -140,8 +141,7 @@ def save_annotations(save_path, annotations, annotation_type):
         if os.path.exists(full_save_path):
             logger.warning('File {} already exists! If you want to recompute annotations for this file, '
                            'please delete it first!'.format(full_save_path))
-            #TODO, currently outputs file until one exists and then fail, resulting in an unknown number being written and similarly not written, maybe should be 'continue' for now.
-            return
+            continue
 
         annotation_group_data = annotation_group_data.drop(columns=
                                            ['raw_filename',