input shouldn't be sliced with expanded slices

5 years ago · 1d9c9644ee
parent 56a1349959
commit 1d9c9644ee
1 changed files with 3 additions and 1 deletions
--- a/fdfd_tools/waveguide_mode.py
+++ b/fdfd_tools/waveguide_mode.py
@ -479,7 +479,9 @@ def expand_wgmode_e(E: field_t,
    slices_exp[axis] = slice(E[0].shape[axis])
    slices_exp = (slice(3), *slices_exp)

-    Ee[slices_exp] = phase_E * numpy.array(E)[slices_Exp]
+    slices_in = tuple(slice(3), *slices)
+
+    Ee[slices_exp] = phase_E * numpy.array(E)[slices_in]

    return Ee