Browse Source

do not skip files

Loann Peurey 5 months ago
parent
commit
c16ead672a
1 changed files with 2 additions and 2 deletions
  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',