I am working with heterogeneous graphs. I have nodes of type A and B. They are connected with edge type ab. I want to use B and its A neighbors(use gat) to update the ab link between them. Cause I need all of the A neighbors of B to use GAT, I update through node functions to B to get all its neighbors. But I cannot find a way to update the edge feature through the node update functions.