"""DST-GNN package for manuscript-aligned temporal COSTE graph analysis."""

from .data import GROUP_TO_STAGE, STAGE_ORDER, TemporalGraphBundle, build_temporal_graphs
from .explain import explain_node_transition, rank_edges_by_stage_change, rank_nodes_by_embedding_drift
from .model import DSTGNN, train_model

__all__ = [
    "DSTGNN",
    "GROUP_TO_STAGE",
    "STAGE_ORDER",
    "TemporalGraphBundle",
    "build_temporal_graphs",
    "explain_node_transition",
    "rank_edges_by_stage_change",
    "rank_nodes_by_embedding_drift",
    "train_model",
]