Fixes LFPy's example_MEA.py

LFPy · Apr 1, 2022 · 239f175 · 239f175
1 parent f739ae1
commit 239f175
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lfpykit/lfpcalc.py b/lfpykit/lfpcalc.py
@@ -988,7 +988,7 @@ def _omega(a_z):
     dL2 = dx2 + dy2
     # avoid denominator approaching 0
     inds = (dL2 + dz2) < (r_limit[rootinds] * r_limit[rootinds])
-    dL2[inds] = r_limit[inds] * r_limit[inds] - dz2[inds]
+    dL2[inds] = r_limit[rootinds][inds] * r_limit[rootinds][inds] - dz2[inds]
 
     def _omega(dz):
         return 1 / np.sqrt(dL2 + dz * dz)