fix: update_mpi needs only zmin as argument, not vector z

Author: elenafuengar

wakis/sources.py

@@ -61,13 +61,13 @@ def update(self, solver, t):
         # update 
         solver.J[self.ixs,self.iys,:,'z'] = self.q*self.v*profile/solver.dx/solver.dy
 
-    def update_mpi(self, solver, t, zmin, z):
+    def update_mpi(self, solver, t, zmin):
         if self.is_first_update:
             self.ixs, self.iys = np.abs(solver.x-self.xsource).argmin(), np.abs(solver.y-self.ysource).argmin()
             self.is_first_update = False
         # reference shift
         s0 = zmin - self.v*self.ti
-        s = z - self.v*t
+        s = solver.z - self.v*t
         # gaussian
         profile = 1/np.sqrt(2*np.pi*self.sigmaz**2)*np.exp(-(s-s0)**2/(2*self.sigmaz**2))
         # update