@@ -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
205207def 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