Skip to content

Commit 60d032c

Browse files
committed
fix attempt 2
1 parent 124893a commit 60d032c

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/spikeinterface/extractors/phykilosortextractors.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,8 @@ def __init__(
149149
del cluster_info["id"]
150150

151151
if remove_empty_units:
152-
cluster_info = cluster_info.query(f"cluster_id in {list(unique_unit_ids)}")
152+
unique_unit_ids_list = [int(clust) for clust in unique_unit_ids]
153+
cluster_info = cluster_info.query(f"cluster_id in {unique_unit_ids_list}")
153154

154155
# update spike clusters and times values
155156
bad_clusters = [clust for clust in unique_unit_ids if clust not in cluster_info["cluster_id"].values]

0 commit comments

Comments
 (0)