Sub-Directories: kinds, parameters

src/pals/__init__.py

@@ -1,22 +1,8 @@
 """Top-level package for PALS.
 
 Re-export commonly used classes from submodules so callers can use
-simpler import statements like `from pals import Drift` instead of
-`from pals.Drift import Drift`.
+simpler import statements like `from pals import Drift`.
 """
 
-from .BaseElement import BaseElement
-from .BeamLine import BeamLine
-from .Drift import Drift
-from .MagneticMultipoleParameters import MagneticMultipoleParameters
-from .Quadrupole import Quadrupole
-from .ThickElement import ThickElement
-
-__all__ = [
-    "BaseElement",
-    "BeamLine",
-    "Drift",
-    "MagneticMultipoleParameters",
-    "Quadrupole",
-    "ThickElement",
-]
+from .kinds import *  # noqa
+from .parameters import *  # noqa

src/pals/Quadrupole.py → src/pals/kinds/Quadrupole.py

@@ -1,7 +1,7 @@
 from typing import Literal
 
 from .ThickElement import ThickElement
-from .MagneticMultipoleParameters import MagneticMultipoleParameters
+from ..parameters import MagneticMultipoleParameters
 
 
 class Quadrupole(ThickElement):

src/pals/kinds/__init__.py

@@ -0,0 +1,9 @@
+"""Re-export commonly used classes from submodules so callers can use
+simpler import statements like `from pals import Drift`.
+"""
+
+from .BaseElement import BaseElement  # noqa: F401
+from .BeamLine import BeamLine  # noqa: F401
+from .Drift import Drift  # noqa: F401
+from .Quadrupole import Quadrupole  # noqa: F401
+from .ThickElement import ThickElement  # noqa: F401

src/pals/parameters/__init__.py

@@ -0,0 +1,5 @@
+"""Re-export commonly used classes from submodules so callers can use
+simpler import statements like `from pals import MagneticMultipoleParameters`.
+"""
+
+from .MagneticMultipoleParameters import MagneticMultipoleParameters  # noqa: F401