Skip to content

Commit 633f745

Browse files
committed
add probe_type in electrode_layouts
1 parent 8f6e280 commit 633f745

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

element_array_ephys/probe.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,13 @@ def create_neuropixels_probe(probe_type: str = "neuropixels 1.0 - 3A"):
143143
}
144144

145145
probe_type = {"probe_type": probe_type}
146-
electrodes = build_electrode_layouts({**neuropixels_probes_config[probe_type["probe_type"]], **probe_type})
146+
electrode_layouts: list[dict] = build_electrode_layouts(
147+
**{**neuropixels_probes_config[probe_type["probe_type"]], **probe_type}
148+
)
147149
with ProbeType.connection.transaction:
148150
ProbeType.insert1(probe_type, skip_duplicates=True)
149151
ProbeType.Electrode.insert(
150-
[{**probe_type, **e} for e in electrodes], skip_duplicates=True
152+
[{**e} for e in electrode_layouts], skip_duplicates=True
151153
)
152154

153155

@@ -203,7 +205,7 @@ class Electrode(dj.Part):
203205

204206

205207
def build_electrode_layouts(
206-
probe_type : str,
208+
probe_type: str,
207209
site_count_per_shank: int,
208210
col_spacing: float = None,
209211
row_spacing: float = None,
@@ -243,7 +245,7 @@ def build_electrode_layouts(
243245
shank_rows = np.repeat(range(row_count), col_count_per_shank)
244246

245247
electrode_layouts = [
246-
{
248+
{
247249
"probe_type": probe_type,
248250
"electrode": (site_count_per_shank * shank_no) + e_id,
249251
"shank": shank_no,

0 commit comments

Comments
 (0)