Using samplers with hypergraph datasets?

[SauravMaheshkar]

I contributed a hypergraph dataset to PyG in #9090; How do I sample this dataset using a sampler? Or create a data loader in any way?

Snippet | Colab

from torch_geometric.loader import NeighborLoader
from torch_geometric.datasets.cornell import CornellTemporalHyperGraphDataset

train_dataset = CornellTemporalHyperGraphDataset(root = "./", name="NDC-classes-25", split="train")
data = train_dataset[0]
kwargs = {'batch_size': 1024, 'num_workers': 6, 'persistent_workers': True}

train_loader = NeighborLoader(
    data,
    input_nodes=None,
    num_neighbors=[-1],
    shuffle=True,
    **kwargs
)

Related: #7365