cugraph_pyg.sampler.io.BufferedSampleReader

class cugraph_pyg.sampler.io.BufferedSampleReader(

nodes_call_groups: list[Tensor],

sample_fn: Callable[[...], Iterator[Tuple[Dict[str, Tensor], int, int]]],

*args,

**kwargs,

)

__init__(

nodes_call_groups: list[Tensor],

sample_fn: Callable[[...], Iterator[Tuple[Dict[str, Tensor], int, int]]],

*args,

**kwargs,

)

Methods

__init__(nodes_call_groups, sample_fn, ...)