"""
Surrogate models for FEA acceleration.

Available surrogates:
- EnsembleSurrogate: Multiple MLPs with uncertainty quantification
- OODDetector: Out-of-distribution detection
"""

from .ensemble_surrogate import (
    EnsembleSurrogate,
    OODDetector,
    create_and_train_ensemble
)

__all__ = [
    'EnsembleSurrogate',
    'OODDetector',
    'create_and_train_ensemble'
]
