From 2b11f49dd2bf945efced1aa073c2b9a6d1e80c80 Mon Sep 17 00:00:00 2001
From: ax3l <ax3l@users.noreply.github.com>
Date: Mon, 6 May 2024 16:06:59 +0000
Subject: [PATCH] Update Stub Files

---
 .../impactx/impactx_pybind/__init__.pyi       |  1 +
 .../impactx/impactx_pybind/elements.pyi       | 25 +++++++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/src/python/impactx/impactx_pybind/__init__.pyi b/src/python/impactx/impactx_pybind/__init__.pyi
index afb4fb197..0861cb3f7 100644
--- a/src/python/impactx/impactx_pybind/__init__.pyi
+++ b/src/python/impactx/impactx_pybind/__init__.pyi
@@ -739,6 +739,7 @@ def push(
         | elements.SoftSolenoid
         | elements.SoftQuadrupole
         | elements.Sol
+        | elements.TaperedPL
         | elements.ThinDipole
     ),
     step: int = 0,
diff --git a/src/python/impactx/impactx_pybind/elements.pyi b/src/python/impactx/impactx_pybind/elements.pyi
index 6a2d57eb1..bed1a9761 100644
--- a/src/python/impactx/impactx_pybind/elements.pyi
+++ b/src/python/impactx/impactx_pybind/elements.pyi
@@ -37,6 +37,7 @@ __all__ = [
     "SoftQuadrupole",
     "SoftSolenoid",
     "Sol",
+    "TaperedPL",
     "Thick",
     "Thin",
     "ThinDipole",
@@ -570,6 +571,7 @@ class KnownElementsList:
             | SoftSolenoid
             | SoftQuadrupole
             | Sol
+            | TaperedPL
             | ThinDipole
         ),
     ) -> None: ...
@@ -610,6 +612,7 @@ class KnownElementsList:
             | SoftSolenoid
             | SoftQuadrupole
             | Sol
+            | TaperedPL
             | ThinDipole
         ),
     ) -> None:
@@ -923,6 +926,28 @@ class Sol(Thick, Alignment):
         Push first the reference particle, then all other particles.
         """
 
+class TaperedPL(Thin, Alignment):
+    def __init__(
+        self,
+        k: float,
+        taper: float,
+        units: int = 0,
+        dx: float = 0,
+        dy: float = 0,
+        rotation: float = 0,
+    ) -> None:
+        """
+        A thin nonlinear plasma lens with transverse taper.
+        """
+
+    def __repr__(self) -> str: ...
+    def push(
+        self, pc: impactx.impactx_pybind.ImpactXParticleContainer, step: int = 0
+    ) -> None:
+        """
+        Push first the reference particle, then all other particles.
+        """
+
 class Thick:
     def __init__(self, ds: float, nslice: float = 1) -> None:
         """