"""WiSER: a wireless scene encoder for radiomap and CIR prediction.

The public API uses CIR terminology. Some internal state-dict keys retain the
historical ``csi_*`` prefix so that the released checkpoint remains loadable.
"""

from .data.dataset import TriplePathDataset
from .engine.trainer import CsiSetTrainer
from .models.model import SparseCsiDetrModel
from .alt_models import FreezeConfig, JointRadiomapCIRModel

__version__ = "0.0.1.dev0"

CIRSetTrainer = CsiSetTrainer
SparseCIRDetrModel = SparseCsiDetrModel

__all__ = [
    "__version__",
    "CIRSetTrainer",
    "FreezeConfig",
    "JointRadiomapCIRModel",
    "CsiSetTrainer",
    "SparseCIRDetrModel",
    "SparseCsiDetrModel",
    "TriplePathDataset",
]