diff --git a/pgl/layers/conv.py b/pgl/layers/conv.py index c3f1ec177cbc203bb707ed03a8d312f3529cf26c..6973b3734b17aa9623c11422893da52568851f61 100644 --- a/pgl/layers/conv.py +++ b/pgl/layers/conv.py @@ -21,7 +21,7 @@ from pgl.utils import paddle_helper from pgl import message_passing import numpy as np -__all__ = ['gcn', 'gat', 'gin', 'gaan', 'gen_conv', 'appnp'] +__all__ = ['gcn', 'gat', 'gin', 'gaan', 'gen_conv', 'appnp', 'gcnii'] def gcn(gw, feature, hidden_size, activation, name, norm=None):