Add reduced diagnostics to `picmi.py` (#3475)

* add reduced diagnostics to picmi
* added test of picmi reduced diagnostics
* changes requested during PR review

1 files changed, 9 insertions, 1 deletions

diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py
index 175917ab6..517c74d42 100644
--- a/Python/pywarpx/WarpX.py
+++ b/Python/pywarpx/WarpX.py
@@ -14,7 +14,7 @@ from .Boundary import boundary
 from .Bucket import Bucket
 from .Collisions import collisions, collisions_list
 from .Constants import my_constants
-from .Diagnostics import diagnostics
+from .Diagnostics import diagnostics, reduced_diagnostics
 from .EB2 import eb2
 from .Geometry import geometry
 from .Interpolation import interpolation
@@ -76,6 +76,14 @@ class WarpX(Bucket):
             for species_diagnostic in diagnostic._species_dict.values():
                 argv += species_diagnostic.attrlist()
 
+        reduced_diagnostics.reduced_diags_names = reduced_diagnostics._diagnostics_dict.keys()
+        argv += reduced_diagnostics.attrlist()
+        for diagnostic in reduced_diagnostics._diagnostics_dict.values():
+            diagnostic.species = diagnostic._species_dict.keys()
+            argv += diagnostic.attrlist()
+            for species_diagnostic in diagnostic._species_dict.values():
+                argv += species_diagnostic.attrlist()
+
         return argv
 
     def init(self, mpi_comm=None):