Fix for wrann error when only label_store is passed. #260

briangow · briangow · commit e6151efb783d · 2021-02-23T09:49:12.000-05:00
diff --git a/wfdb/io/annotation.py b/wfdb/io/annotation.py
@@ -280,6 +280,11 @@ def wrann(self, write_fs=False, write_dir=''):
         if 'label_store' not in present_label_fields:
             self.convert_label_attribute(source_field=present_label_fields[0],
                                          target_field='label_store')
+            
+        # Calculate the symbol field if necessary
+        if 'symbol' not in present_label_fields:
+            self.convert_label_attribute(source_field=present_label_fields[0],
+                                         target_field='symbol')
 
         # Write the header file using the specified fields
         self.wr_ann_file(write_fs=write_fs, write_dir=write_dir)
