model_code/__init__.py

# TRIADS model_code package
# Import the model classes for convenience

from .steels_model import DeepHybridTRM as SteelsModel
from .expt_gap_model import DeepHybridTRM as ExptGapModel
from .classification_model import DeepHybridTRM as ClassificationModel
from .jdft2d_model import DeepHybridTRM as Jdft2dModel

__all__ = ["SteelsModel", "ExptGapModel", "ClassificationModel", "Jdft2dModel"]