Does it support the PyG framework?

[jweihe]

If not, how can we adapt PyG data and networks to this framework? Thank you!

[mufeili]

You can try modifying this file: https://github.com/dmlc/GNNLens2/blob/main/python/gnnlens/writer.py