address PR comments

Co-authored-by: Dimitri Yatsenko <[email protected]>
datajoint · Sep 15, 2023 · f42f1fc · f42f1fc
1 parent 452ff31
commit f42f1fc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/element_array_ephys/readers/spikeglx.py b/element_array_ephys/readers/spikeglx.py
@@ -450,12 +450,12 @@ def _transform_geom_to_shank(self):
 
         from . import probe_geometry
 
-        probe_params = {
-            n: v
-            for n, v in zip(
-                probe_geometry.geom_param_names, probe_geometry.M[self.probe_PN]
+        probe_params = dict(
+            zip(
+                probe_geometry.geom_param_names, 
+                probe_geometry.M[self.probe_PN]
             )
-        }
+        )
         probe_params["probe_type"] = self.probe_PN
         elec_pos_df = probe_geometry.build_npx_probe(**probe_params)