Update Stub Files

ECP-WarpX · May 23, 2024 · 30cbdb5 · 30cbdb5
1 parent 36beb75
commit 30cbdb5
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 3 deletions.
diff --git a/src/python/impactx/__init__.pyi b/src/python/impactx/__init__.pyi
@@ -68,7 +68,7 @@ __author__: str = (
     "Axel Huebl, Chad Mitchell, Ryan Sandberg, Marco Garten, Ji Qiang, et al."
 )
 __license__: str = "BSD-3-Clause-LBNL"
-__version__: str = "24.04"
+__version__: str = "24.05"
 s: impactx_pybind.CoordSystem  # value = <CoordSystem.s: 0>
 t: impactx_pybind.CoordSystem  # value = <CoordSystem.t: 1>
 cxx = impactx_pybind
diff --git a/src/python/impactx/impactx_pybind/__init__.pyi b/src/python/impactx/impactx_pybind/__init__.pyi
@@ -757,6 +757,6 @@ __author__: str = (
     "Axel Huebl, Chad Mitchell, Ryan Sandberg, Marco Garten, Ji Qiang, et al."
 )
 __license__: str = "BSD-3-Clause-LBNL"
-__version__: str = "24.04"
+__version__: str = "24.05"
 s: CoordSystem  # value = <CoordSystem.s: 0>
 t: CoordSystem  # value = <CoordSystem.t: 1>
diff --git a/src/python/impactx/impactx_pybind/elements.pyi b/src/python/impactx/impactx_pybind/elements.pyi
@@ -577,7 +577,38 @@ class KnownElementsList:
     ) -> None: ...
     @typing.overload
     def __init__(self, arg0: list) -> None: ...
-    def __iter__(self) -> typing.Iterator: ...
+    def __iter__(
+        self,
+    ) -> typing.Iterator[
+        Empty
+        | Aperture
+        | Buncher
+        | CFbend
+        | ChrAcc
+        | ChrDrift
+        | ChrPlasmaLens
+        | ChrQuad
+        | ConstF
+        | BeamMonitor
+        | DipEdge
+        | Drift
+        | ExactDrift
+        | ExactSbend
+        | Kicker
+        | Multipole
+        | NonlinearLens
+        | Programmable
+        | PRot
+        | Quad
+        | RFCavity
+        | Sbend
+        | ShortRF
+        | SoftSolenoid
+        | SoftQuadrupole
+        | Sol
+        | TaperedPL
+        | ThinDipole
+    ]: ...
     def __len__(self) -> int:
         """
         The length of the list.