Fix proposed by NeuralEnsemble#420. (NeuralEnsemble#421)

INM-6 · Sep 6, 2021 · f5c3e15 · f5c3e15
1 parent 3520ac2
commit f5c3e15
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/elephant/current_source_density_src/icsd.py b/elephant/current_source_density_src/icsd.py
@@ -187,12 +187,13 @@ def __init__(self, lfp, coord_electrode, **kwargs):
         if self.vaknin_el:
             # extend lfps array by duplicating potential at endpoint contacts
             if lfp.ndim == 1:
-                self.lfp = np.empty((lfp.shape[0] + 2, )) * lfp.units
+                self.lfp = np.empty((lfp.shape[0] + 2,))
             else:
-                self.lfp = np.empty((lfp.shape[0] + 2, lfp.shape[1])) * lfp.units
-            self.lfp[0, ] = lfp[0, ]
+                self.lfp = np.empty((lfp.shape[0] + 2, lfp.shape[1]))
+            self.lfp[0,] = lfp[0,]
             self.lfp[1:-1, ] = lfp
-            self.lfp[-1, ] = lfp[-1, ]
+            self.lfp[-1,] = lfp[-1,]
+            self.lfp = self.lfp * lfp.units
         else:
             self.lfp = lfp