diff --git a/element_array_ephys/probe.py b/element_array_ephys/probe.py index 8157c022..1d517cf3 100644 --- a/element_array_ephys/probe.py +++ b/element_array_ephys/probe.py @@ -107,13 +107,10 @@ def create_neuropixels_probe(probe_type: str = "neuropixels 1.0 - 3A"): npx_probes_config["neuropixels 2.0 - MS"] = npx_probes_config["NP2010"] probe_type = {"probe_type": probe_type} - probe_params = { - n: v - for n, v in zip( + probe_params = dict(zip( probe_geometry.geom_param_names, - npx_probes_config[probe_type["probe_type"]], - ) - } + npx_probes_config[probe_type["probe_type"]] + )) electrode_layouts = probe_geometry.build_npx_probe( **{**probe_params, **probe_type} )