From 452ff31c952f641f329771c22c39a4e6845d7588 Mon Sep 17 00:00:00 2001 From: Thinh Nguyen Date: Fri, 15 Sep 2023 13:45:59 -0500 Subject: [PATCH] address PR comments Co-authored-by: Dimitri Yatsenko --- element_array_ephys/probe.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) 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} )