from .convs import GCNConv, GATConv, GINConv
# from .convs import ShrinkGCNConv, ShrinkGATConv, ShrinkGINConv
# from .convs import ShrinkHelper
# from .basic_gnn import BasicGNN, BasicLayerOptions, BasicInputOptions, BasicStraightOptions

# class ShrinkGCN(BasicGNN):
#     def init_conv(self, in_channels: int, out_channels: int, **kwargs):
#         return ShrinkGCNConv(in_channels, out_channels, **kwargs)
    

# class ShrinkGAT(BasicGNN):
#     def init_conv(self, in_channels: int, out_channels: int, **kwargs):
#         return ShrinkGATConv(in_channels, out_channels, **kwargs)
    
# class ShrinkGIN(BasicGNN):
#     def init_conv(self, in_channels: int, out_channels: int, **kwargs):
#         return ShrinkGINConv(in_channels, out_channels, **kwargs)
    