Example code of Node Flow in Graph Attention Neural Network

I would like to build a Graph Attention Network in Node Flow mode because of large amount of data.
I found example about GAT and the NodeFlow tutorials. There is “edge_attention” in the GAT code, but in the nodeflow block_compute api, there is no argument for edge_attention function.
Should I insert the edge_attention function into the message_func ???

Sorry for the late reply. I’ve prepared a demo here.