How to use modulus with torch_geometric datasets

Hello,

In Modulus Datapipes datasets are written in DGL structure. I am wondering if it possible to write a custom dataset following torch geometric structure or import a dataset from torch_geometric.datasets Link

Thank you