code: def test_epoch_idx():
      batch_size = 3
      epoch_size = 4
      for batch_info in (True, False):
          batch_cb = BatchCb(batch_info, batch_size, epoch_size)
          yield _test_epoch_idx, batch_size, epoch_size, batch_cb, batch_info, True
      sample_cb = SampleCb(batch_size, epoch_size)
      yield _test_epoch_idx, batch_size, epoch_size, sample_cb, None, False
  
imports: import random
  import nvidia.dali.fn
  import nvidia.dali.ops
  import nvidia.dali.types
  import numpy
  from nvidia.dali import Pipeline, pipeline_def
  from test_utils import check_batch
  from nose_utils import raises, assert_warns, assert_raises
  from nvidia.dali.types import DALIDataType
  from numpy.random import default_rng
label_desc: Use image node_id 11 for calling a global function with example usage: test_epoch_idx() without return types