model/__init__.py

from .Transolver2D import Transolver2D
from .Transolver2D_plus import Transolver2D_plus

__all__ = ["Transolver2D", "Transolver2D_plus"]