model/egnn/__init__.py

from .egnn_pytorch import EGNN, EGNN_Network
from .egnn_pytorch_geometric import EGNN_Sparse, EGNN_Sparse_Network