NeighborSampler depth

Hi everyone,

I want to sample with data loader neighbors up to a given depth like so (K=2):
Screenshot 2022-09-21 at 12.59.17

How can I achieve this? NeighborSampler from what I understand you can only specify the number of neighbors per GNN layer, but not the neighbourhood depth.