# graph neural network model

from torch_geometric.nn import GNN

class HAIPRGNN(GNN):
    def __init__(self, config):
        super().__init__(config)

    def forward(self, x, edge_index, edge_attr):
        return self.gnn(x, edge_index, edge_attr)
