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

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

classes = __all__
