address PR comments

Co-authored-by: Dimitri Yatsenko <[email protected]>
kabilar · Sep 15, 2023 · 452ff31 · 452ff31
1 parent 4407678
commit 452ff31
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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}
         )