docstr: Parameters
  ----------
  data: CuGraphStore or (CuGraphStore, CuGraphStore)
      The CuGraphStore or stores where the graph/feature data is held.
  
  batch_size: int (required)
      The number of input nodes in each batch.
  
  input_nodes: Union[InputNodes, int] (required)
      The input nodes associated with this sampler.
  
  **kwargs: kwargs
      Keyword arguments to pass through for sampling.
      i.e. "shuffle", "fanout"
      See BulkSampleLoader.
docstr_tok: ["Parameters","--","--","--","--","--","data",":","CuGraphStore","or","(","CuGraphStore",",","CuGraphStore",")","The","CuGraphStore","or","stores","where","the","graph\/feature","data","is","held",".","batch_size",":","int","(","required",")","The","number","of","input","nodes","in","each","batch",".","input_nodes",":","Union","[","InputNodes",",","int","]","(","required",")","The","input","nodes","associated","with","this","sampler",".","*","*","kwargs",":","kwargs","Keyword","arguments","to","pass","through","for","sampling",".","i.e",".","``","shuffle","''",",","``","fanout","''","See","BulkSampleLoader","."]
code: def __init__(
      self,
      data: Union[CuGraphStore, Tuple[CuGraphStore, CuGraphStore]],
      input_nodes: Union[InputNodes, int] = None,
      batch_size: int = None,
      **kwargs,
  ):
      """
      Parameters
      ----------
      data: CuGraphStore or (CuGraphStore, CuGraphStore)
          The CuGraphStore or stores where the graph/feature data is held.
  
      batch_size: int (required)
          The number of input nodes in each batch.
  
      input_nodes: Union[InputNodes, int] (required)
          The input nodes associated with this sampler.
  
      **kwargs: kwargs
          Keyword arguments to pass through for sampling.
          i.e. "shuffle", "fanout"
          See BulkSampleLoader.
      """
  
      if input_nodes is None:
          raise ValueError("input_nodes is required")
      if batch_size is None:
          raise ValueError("batch_size is required")
  
      # Allow passing in a feature store and graph store as a tuple, as
      # in the standard PyG API.  If only one is passed, it is assumed
      # it is behaving as both a graph store and a feature store.
      if isinstance(data, (list, tuple)):
          self.__feature_store, self.__graph_store = data
      else:
          self.__feature_store = data
          self.__graph_store = data
  
      self.__batch_size = batch_size
      self.__input_nodes = input_nodes
      self.inner_loader_args = kwargs
  
file_path: cugraph/python/cugraph-pyg/cugraph_pyg/loader/cugraph_node_loader.py
imports: import tempfile
  import os
  import re
  import warnings
  import cupy
  import cudf
  from cugraph.gnn import BulkSampler
  from cugraph.utilities.utils import import_optional, MissingModule
  from cugraph_pyg.data import CuGraphStore
  from cugraph_pyg.sampler.cugraph_sampler import _sampler_output_from_sampling_results_heterogeneous, _sampler_output_from_sampling_results_homogeneous_csr, _sampler_output_from_sampling_results_homogeneous_coo, filter_cugraph_store_csc
  from typing import Union, Tuple, Sequence, List, Dict
label_desc: Use image node_id 1 to create a new CuGraphNeighborLoader object  with example: obj = CuGraphNeighborLoader(data, input_nodes, batch_size)