Skip to content

Commit 92c8d05

Browse files
committed
Make elec_ids and shank_ids int
1 parent 2ffd8cf commit 92c8d05

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/probeinterface/neuropixels_tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -880,11 +880,11 @@ def read_openephys(
880880
contact_per_shank = pt_metadata["cols_per_shank"] * pt_metadata["rows_per_shank"]
881881

882882
if num_shank == 1:
883-
elec_ids = np.arange(contact_per_shank)
883+
elec_ids = np.arange(contact_per_shank, dtype=int)
884884
shank_ids = None
885885
else:
886-
elec_ids = np.concatenate([np.arange(contact_per_shank) for i in range(num_shank)])
887-
shank_ids = np.concatenate([np.zeros(contact_per_shank) + i for i in range(num_shank)])
886+
elec_ids = np.concatenate([np.arange(contact_per_shank, dtype=int) for i in range(num_shank)])
887+
shank_ids = np.concatenate([np.zeros(contact_per_shank, dtype=int) + i for i in range(num_shank)])
888888

889889
full_probe = _make_npx_probe_from_description(
890890
pt_metadata, probe_part_number, elec_ids, shank_ids, mux_info=mux_info

0 commit comments

Comments
 (0)