matdeeplearn/models/__init__.py

from .gcn import GCN
from .mpnn import MPNN
from .schnet import SchNet
from .cgcnn import CGCNN
from .megnet import MEGNet
from .descriptor_nn import SOAP, SM

__all__ = [
    "GCN",
    "MPNN",
    "SchNet",
    "CGCNN",
    "MEGNet",
    "SOAP",
    "SM",
]