diff --git a/element_array_ephys/readers/probe_geometry.py b/element_array_ephys/readers/probe_geometry.py index 11e3ae99..b0f27765 100644 --- a/element_array_ephys/readers/probe_geometry.py +++ b/element_array_ephys/readers/probe_geometry.py @@ -188,7 +188,7 @@ def build_electrode_layouts( row_offset = np.zeros_like(x_coords) else: assert len(row_offset) == row_count - row_offset = np.tile(row_offset, col_count_per_shank) + row_offset = np.repeat(row_offset, col_count_per_shank) x_coords = x_coords + row_offset shank_cols = np.tile(range(col_count_per_shank), row_count)