Add missing return

5 years ago · 56a1349959
parent f2d061c921
commit 56a1349959
1 changed files with 2 additions and 0 deletions
--- a/fdfd_tools/waveguide_mode.py
+++ b/fdfd_tools/waveguide_mode.py
@ -425,6 +425,8 @@ def compute_source_wg(E: field_t,
                         omega=omega, dxes=dxes, axis=axis,
                         polarity=polarity, slices=slices4,
                         epsilon=epsilon, mu=mu)
+    return J
+

 def compute_overlap_ce(E: field_t,
                       wavenumber: complex,