Source code for pyaml.magnet.skewquad

from ..lattice.polynom_info import PolynomInfo
from .magnet import Magnet, MagnetConfigModel

# Define the main class name for this module
PYAMLCLASS = "SkewQuad"




class ConfigModel(MagnetConfigModel):
    """Configuration model for SkewQuad magnet."""

    ...





class SkewQuad(Magnet):
    """SkewQuad class"""

    polynom = PolynomInfo("PolynomA", 1)

    def __init__(self, cfg: ConfigModel):
        super().__init__(
            cfg.name,
            cfg.model if hasattr(cfg, "model") else None,
        )
        self._cfg = cfg