from .contrastive.model import Contrastive, GraphCL, GRACE, InfoGraph, MVGRL, NodeMVGRL, pGRACE

__all__ = [
    'Contrastive',
    'GraphCL',
    'GRACE',
    'InfoGraph',
    'MVGRL',
    'NodeMVGRL',
    'pGRACE'
]

classes = __all__
