code: def get_batch_id_series(n_output_rows: int, batch_size: int):
      num_batches = (n_output_rows + batch_size - 1) // batch_size
      print(f"Number of batches = {num_batches}".format(num_batches))
      batch_ar = cp.arange(0, num_batches).repeat(batch_size)
      batch_ar = batch_ar[0:n_output_rows].astype(cp.int32)
      return cudf.Series(batch_ar)
  
file_path: cugraph/python/cugraph-dgl/cugraph_dgl/dataloading/dataloader.py
imports: from __future__ import annotations
  import os
  import shutil
  import cugraph_dgl
  import cupy
  import cudf
  from cugraph.utilities.utils import import_optional
  from cugraph.gnn import BulkSampler
  from dask.distributed import default_client, Event
  from cugraph_dgl.dataloading import HomogenousBulkSamplerDataset, HeterogenousBulkSamplerDataset
  from cugraph_dgl.dataloading.utils.extract_graph_helpers import create_cugraph_graph_from_edges_dict
label_desc: Use image node_id 1 for calling a global function with example usage: get_batch_id_series(n_output_rows, batch_size) and returns: cudf