code: def test_jax_peekable_iterator_with_variable_shapes_pipeline():
      # given
      batch_size = 1
      pipe = pipeline_with_variable_shape_output(batch_size)
  
      iterator = DALIPeekableIterator(
          [pipe], ["data"], size=batch_size * 100
      )
      iterator.next()
  
      # when
      iterator.next()
  
file_path: DALI/dali/test/python/jax_plugin/test_peekable_iterator.py
imports: import jax.numpy
  from nvidia.dali.pipeline import pipeline_def
  from nvidia.dali.plugin.jax.clu import DALIGenericPeekableIterator
  from nvidia.dali.plugin.jax.iterator import DALIGenericIterator
  from utils import iterator_function_def
  from clu.data.dataset_iterator import ArraySpec
  from nose_utils import raises
  import time
  import inspect
  from utils import pipeline_with_variable_shape_output
label_desc: Use image node_id 4 for calling a global function with example usage: test_jax_peekable_iterator_with_variable_shapes_pipeline() without return types