Fixes Pandas .loc update for annotations #255

Lucas-Mc · Lucas-Mc · commit f9e797d9921b · 2020-09-29T14:16:53.000-04:00
Fixes the deprecation of .loc being used with missing labels and replaces it with the suggested .reindex(). Fixes #255. Fixes TravisCI error
diff --git a/wfdb/io/annotation.py b/wfdb/io/annotation.py
@@ -1204,7 +1204,10 @@ def convert_label_attribute(self, source_field, target_field, inplace=True,
         label_map = self.create_label_map(inplace=False)
         label_map.set_index(label_map[source_field].values, inplace=True)
 
-        target_item = label_map.loc[getattr(self, source_field), target_field].values
+        try:
+            target_item = label_map.loc[getattr(self, source_field), target_field].values
+        except KeyError:
+            target_item = label_map.reindex(index=getattr(self, source_field), columns=[target_field]).values.flatten()
 
         if target_field != 'label_store':
             # Should already be int64 dtype if target is label_store
