import torch_spline_conv as SplineConv