"""Physics modules for DiffTB."""

from difftb.physics.bands import BandStructure, compute_bands
from difftb.physics.soc import add_soc, soc_matrix
from difftb.physics.spin import SpinConfig, add_spin_polarization

__all__ = [
    "SpinConfig",
    "add_spin_polarization",
    "add_soc",
    "soc_matrix",
    "compute_bands",
    "BandStructure",
]
