The previous datasets do not use edge features. As as mentioned in Section , this is a straightforward extension (Appendix ). We test this extension on the OGB-MolHIV molecule classification dataset, which uses edge features [1]}. In Table REF  we can see that AgentNet performs well in this scenario and outperforms even the best ESAN model.
