Convert Train/Test split in RGCN for heterograph

Could you please provide an explanation and description of the parameters and their types used in the generate_sampled_graph_and_labels() function?

I am trying to manipulate this to do sampling on a heterograph. Do you think that the batched heterograph is a good solution here? I am planning on constructing a sampled heterograph for each edge type and then merging them together.

Thanks for your help in advance!