Graph Neural Network Python