Final Review & Cleanup

Author: ax3l

src/pals/kinds/BeamLine.py

@@ -53,40 +53,39 @@ class BeamLine(BaseElement):
                 # Base classes (for testing compatibility)
                 BaseElement,
                 ThickElement,
-                # Original elements
-                Drift,
-                Quadrupole,
+                # User-Facing element kinds
                 "BeamLine",
-                # New schema elements
-                Marker,
-                Sextupole,
-                Octupole,
-                Multipole,
-                RBend,
-                SBend,
-                Solenoid,
-                RFCavity,
-                Patch,
-                FloorShift,
-                Fork,
+                ACKicker,
                 BeamBeam,
                 BeginningEle,
-                Fiducial,
-                NullEle,
-                Kicker,
-                ACKicker,
+                Converter,
                 CrabCavity,
+                Drift,
                 EGun,
                 Feedback,
+                Fiducial,
+                FloorShift,
+                Foil,
+                Fork,
                 Girder,
                 Instrument,
+                Kicker,
+                Marker,
                 Mask,
                 Match,
+                Multipole,
+                NullEle,
+                Octupole,
+                Patch,
+                Quadrupole,
+                RBend,
+                RFCavity,
+                SBend,
+                Sextupole,
+                Solenoid,
                 Taylor,
-                Wiggler,
-                Converter,
-                Foil,
                 UnionEle,
+                Wiggler,
             ],
             Field(discriminator="kind"),
         ]

src/pals/kinds/Converter.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .BaseElement import BaseElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/CrabCavity.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/EGun.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Instrument.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Kicker.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Mask.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Multipole.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Octupole.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction
 
 

src/pals/kinds/Sextupole.py

@@ -1,8 +1,7 @@
 from typing import Literal, Optional
 
 from .ThickElement import ThickElement
-from ..parameters import ElectricMultipoleParameters
-from ..parameters import MagneticMultipoleParameters
+from ..parameters import ElectricMultipoleParameters, MagneticMultipoleParameters
 from ._warnings import under_construction