"""HiCon-GNN publication package."""

from .data import ConcreteSample, MultiscaleBatch, collate_multiscale
from .models.hierarchical import IntegratedMultiscaleModel, TabularANN
from .real_data import (
    TABULAR_DIM,
    TARGET_NAME,
    ConcreteMixDataset,
    Standardizer,
    StrengthHead,
    collect_strength_predictions,
    evaluate_strength,
    fit_encoder_standardizers,
    metrics,
    raw_mix_vector,
    read_table,
    strength_column,
)
from .schema import DEFAULT_SCHEMA, SchemaSpec
from .splits import load_split_indices, random_split_indices, split_indices_sha256
from .visualize import (
    plot_parity,
    plot_r2_bars,
    plot_rve_graph,
    plot_training_curves,
)

__all__ = [
    "ConcreteMixDataset",
    "ConcreteSample",
    "DEFAULT_SCHEMA",
    "IntegratedMultiscaleModel",
    "MultiscaleBatch",
    "SchemaSpec",
    "Standardizer",
    "StrengthHead",
    "TABULAR_DIM",
    "TARGET_NAME",
    "TabularANN",
    "collect_strength_predictions",
    "collate_multiscale",
    "evaluate_strength",
    "fit_encoder_standardizers",
    "metrics",
    "load_split_indices",
    "random_split_indices",
    "plot_parity",
    "plot_r2_bars",
    "plot_rve_graph",
    "plot_training_curves",
    "raw_mix_vector",
    "read_table",
    "strength_column",
    "split_indices_sha256",
]
