Fix for pandas 3.0

larsoner · larsoner · commit b23f17b3e1aa · 2026-01-23T10:14:26.000-05:00
diff --git a/wfdb/io/annotation.py b/wfdb/io/annotation.py
@@ -1469,7 +1469,7 @@ def convert_label_attribute(
                 return
 
         label_map = self.create_label_map(inplace=False)
-        label_map.set_index(label_map[source_field].values, inplace=True)
+        label_map.set_index(keys=pd.Index(label_map[source_field].values), inplace=True)
 
         try:
             target_item = label_map.loc[
@@ -3057,7 +3057,11 @@ def __init__(self, extension, description, human_reviewed):
         "human_reviewed": [ac.human_reviewed for ac in ann_classes],
     }
 )
-ann_class_table.set_index(ann_class_table["extension"].values, inplace=True)
+ann_class_table.set_index(
+    keys=pd.Index(ann_class_table["extension"].values),
+    inplace=True,
+)
+print(ann_class_table)
 ann_class_table = ann_class_table[
     ["extension", "description", "human_reviewed"]
 ]

Original file line number	Diff line number	Diff line change
`@@ -1469,7 +1469,7 @@ def convert_label_attribute(`
`1469`	`1469`	`return`
`1470`	`1470`
`1471`	`1471`	`label_map = self.create_label_map(inplace=False)`
`1472`		`- label_map.set_index(label_map[source_field].values, inplace=True)`
	`1472`	`+ label_map.set_index(keys=pd.Index(label_map[source_field].values), inplace=True)`
`1473`	`1473`
`1474`	`1474`	`try:`
`1475`	`1475`	`target_item = label_map.loc[`
`@@ -3057,7 +3057,11 @@ def __init__(self, extension, description, human_reviewed):`
`3057`	`3057`	`"human_reviewed": [ac.human_reviewed for ac in ann_classes],`
`3058`	`3058`	`}`
`3059`	`3059`	`)`
`3060`		`-ann_class_table.set_index(ann_class_table["extension"].values, inplace=True)`
	`3060`	`+ann_class_table.set_index(`
	`3061`	`+ keys=pd.Index(ann_class_table["extension"].values),`
	`3062`	`+ inplace=True,`
	`3063`	`+)`
	`3064`	`+print(ann_class_table)`
`3061`	`3065`	`ann_class_table = ann_class_table[`
`3062`	`3066`	`["extension", "description", "human_reviewed"]`
`3063`	`3067`	`]`