Merge pull request #460 from ajadczaksunriselabs/pandas-set-indexer

bug-fix: Pandas set indexing error
MIT-LCP · Jul 6, 2023 · 2cdb44a · 2cdb44a
2 parents c0f1b12 + 1737eb0
commit 2cdb44a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wfdb/io/annotation.py b/wfdb/io/annotation.py
@@ -1352,7 +1352,7 @@ def get_contained_labels(self, inplace=True):
         else:
             raise Exception("No annotation labels contained in object")
 
-        contained_labels = label_map.loc[index_vals, :]
+        contained_labels = label_map.loc[list(index_vals), :]
 
         # Add the counts
         for i in range(len(counts[0])):