code: def get_batch_start_end(fn):
      batch_str = fn.split("batch=")[1]
      batch_start, batch_end = batch_str.split("-")
      batch_end = batch_end.split(".parquet")[0]
      return int(batch_start), int(batch_end)
  
imports: from __future__ import annotations
  from typing import Tuple, Dict, Optional, List, Union
  import os
  import cudf
  from cugraph.utilities.utils import import_optional
  from cugraph_dgl.dataloading.utils.sampling_helpers import create_homogeneous_sampled_graphs_from_dataframe, create_heterogeneous_sampled_graphs_from_dataframe, create_homogeneous_sampled_graphs_from_dataframe_csc
label_desc: Use image node_id 2 for calling a global function with example usage: get_batch_start_end(fn) and returns: int,  int