Graph Neural Network Python Code