Skip to content

Commit

Permalink
Merge pull request #3365 from h-mayorquin/read_quality_in_phy_as_string
Browse files Browse the repository at this point in the history
"quality" property to be read as string instead of object in `BasePhyKilosortSortingExtractor`
  • Loading branch information
alejoe91 authored Sep 5, 2024
2 parents 740e779 + 5953d6c commit 0e36a3a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/spikeinterface/extractors/phykilosortextractors.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 0e36a3a

Please sign in to comment.