Implemented multiple antennas in picmi

1 files changed, 5 insertions, 2 deletions

diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py
index 4e4e41187..5f7477e58 100644
--- a/Python/pywarpx/WarpX.py
+++ b/Python/pywarpx/WarpX.py
@@ -5,7 +5,7 @@ from .Geometry import geometry
 from .Algo import algo
 from .Langmuirwave import langmuirwave
 from .Interpolation import interpolation
-from .Laser import laser
+from .Lasers import lasers, lasers_list
 from . import Particles
 from .Particles import particles, particles_list
 
@@ -25,7 +25,7 @@ class WarpX(Bucket):
         argv += langmuirwave.attrlist()
         argv += interpolation.attrlist()
         argv += particles.attrlist()
-        argv += laser.attrlist()
+        argv += lasers.attrlist()
 
         # --- Search through species_names and add any predefined particle objects in the list.
         particles_list_names = [p.instancename for p in particles_list]
@@ -43,6 +43,9 @@ class WarpX(Bucket):
         for particle in particles_list:
             argv += particle.attrlist()
 
+        for laser in lasers_list:
+            argv += laser.attrlist()
+
         return argv
 
     def init(self):