Merge branch 'add_cupy_callbacks_picmi' of https://github.com/oshapoval/WarpX into add_cupy_callbacks_picmi

Author: oshapoval

Examples/Physics_applications/capacitive_discharge/inputs_test_2d_background_mcc_picmi.py

@@ -73,8 +73,6 @@ def __init__(self, grid, **kwargs):
             required_precision=1,
             **kwargs,
         )
-        self.rho_wrapper = None
-        self.phi_wrapper = None
         self.time_sum = 0.0
 
     def solver_initialize_inputs(self):
@@ -160,15 +158,13 @@ def _run_solve(self):
         Poisson's equation."""
 
         # get rho from WarpX
-        if self.rho_wrapper is None:
-            self.rho_wrapper = sim.fields.get("rho_fp", level=0)
-        self.rho_data = self.rho_wrapper[(), ()]
+        rho_wrapper = sim.fields.get("rho_fp", level=0)
+        self.rho_data = rho_wrapper[(), ()]
 
         self.solve()
 
-        if self.phi_wrapper is None:
-            self.phi_wrapper = sim.fields.get("phi_fp", level=0)
-        self.phi_wrapper[(), ()] = self.phi
+        phi_wrapper = sim.fields.get("phi_fp", level=0)
+        phi_wrapper[(), ()] = self.phi[...]
 
     def solve(self):
         """The solution step. Includes getting the boundary potentials and