Could you update the latest version of this work without 'group_apply_edges'?

[MrYaoH]

Dear author,

I'm very interesting about this work. However, the problem, the code require dgl version < 0.4.3, "bother" me, since in current workstations, dgl version usually > 1.0, no speaking of 0.4.3. Therefore, I wonder if you can update the code without 'group_apply_edges'.

Thank you very much!