quality in phy to string

SpikeInterface · Sep 5, 2024 · 5953d6c · 5953d6c
1 parent 740e779
commit 5953d6c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spikeinterface/extractors/phykilosortextractors.py b/src/spikeinterface/extractors/phykilosortextractors.py
@@ -159,7 +159,8 @@ def __init__(
                 self.set_property(key="original_cluster_id", values=cluster_info[prop_name])
             elif prop_name == "group":
                 # rename group property to 'quality'
-                self.set_property(key="quality", values=cluster_info[prop_name])
+                values = cluster_info[prop_name].values.astype("str")
+                self.set_property(key="quality", values=values)
             else:
                 if load_all_cluster_properties:
                     # pandas loads strings with empty values as objects with NaNs