code: def get_batch_one_arg_callback_with_batch_info(
      dtype, iter_limit=1000, batch_size=None, dense=True
  ):
      def callback(x):
          if x.iteration > iter_limit:
              raise StopIteration()
          size = (x.iteration % 16 + 4,)
          result = [
              np.full(size, x.iteration, dtype=dtype)
          ] * batch_size
          for i, elem in enumerate(result):
              elem[0] = i
              elem[1] = x.iteration
              elem[2] = x.epoch_idx
          return np.stack(result) if dense else result
  
      return callback
  
imports: import copy
  import numpy
  import nvidia.dali.plugin.tf
  import tensorflow
  from nvidia.dali import Pipeline, fn
  from test_utils import RandomlyShapedDataIterator
label_desc: Use image node_id 3 for calling a global function with example usage: get_batch_one_arg_callback_with_batch_info(dtype, iter_limit, batch_size, dense) and returns: callback, unknown