from .gcn import GCNModel
from .gat import GATModel
from .sgc import SGCModel
from .gatv2 import GATV2Model
from .gaan import GaANModel
from .graphsage import GraphSAGE_Full_Model, GraphSAGE_Sample_Model
from .gcnii import GCNIIModel
from .appnp import APPNPModel
from .gin import GINModel
from .rgcn import RGCN
from .compgcn import CompGCN
from .agnn import AGNNModel
from .jknet import JKNet
from .han import HAN
from .dgi import DGIModel
from .grace import GraceModel
from .grade import GRADE
from .chebnet import ChebNetModel
from .simplehgn import SimpleHGNModel
from .fagcn import FAGCNModel
from .gprgnn import GPRGNNModel
from .dgcnn import DGCNNModel
from .seal import DGCNN
from .hgt import HGTModel
from .film import FILMModel
from .pna import PNAModel
from .mixhop import MIXHOPModel
from .hcha import HCHA
from .hardgat import HardGATModel
from .mlp import MLP
from .graphgan_generator import Generator
from .graphgan_discriminator import Discriminator
from .graphgan import GraphGAN
from .vgae import VGAEModel, GAEModel
from .gen import GEstimationN
from .skipgram import SkipGramModel
from .deepwalk import DeepWalkModel
from .node2vec import Node2vecModel
from .hpn import HPN
from .gmm import GMMModel
from .herec import HERec
from .metapath2vec import MetaPath2Vec
from .iehgcn import ieHGCNModel
from .tadw import TADWModel
from .mgnni import MGNNI_m_MLP, MGNNI_m_att 
from .magcl import NewGrace


__all__ = [
    'GCNModel',
    'GATModel',
    'GaANModel',
    'SGCModel',
    'GATV2Model',
    'GraphSAGE_Full_Model',
    'GraphSAGE_Sample_Model',
    'GCNIIModel',
    'APPNPModel',
    'GINModel',
    'RGCN',
    'CompGCN',
    'AGNNModel',
    'JKNet',
    'HAN',
    'DGIModel',
    'GraceModel',
    'GRADE',
    'ChebNetModel',
    'SimpleHGNModel',
    'FAGCNModel',
    'GPRGNNModel',
    'DGCNN',
    'MIXHOPModel',
    'HCHA',
    'HGTModel',
    'PNAModel',
    'HardGATModel',
    # 'GraphGAN',
    # 'Generator',
    # 'Discriminator',
    'DGCNNModel',
    'FILMModel',
    'GEstimationN',
    'SkipGramModel',
    'DeepWalkModel',
    'Node2vecModel',
    'VGAEModel',
    'GAEModel',
    'HPN',
    'GMMModel',
    'HERec',
    'MetaPath2Vec'
    'ieHGCNModel',
    'TADWModel',
    'MGNNI_m_MLP', 
    'MGNNI_m_att', 
    'NewGrace',

]

classes = __all__