Good toolkit! In addition, I wonder if AutoGL supports multi-classification edge prediction tasks. Thanks a lot.