Change to directed graph

Hello! I noticed that you used undirected graph in your model, I would like to know what modifications I need to make to the model if I want to use the directed graph dataset for this model? Thanks a lot~