change to dgl.copy_e() https://docs.dgl.ai/en/0.8.x/generated/dgl.function.copy_edge.html#
change to dgl.copy_e() https://docs.dgl.ai/en/0.8.x/generated/dgl.function.copy_edge.html#