code: def test_iterator_last_batch_policy_partial_exception():
      pipe = pipeline_def(iterator_function_def)(
          batch_size=batch_size, num_threads=4, device_id=0
      )
      DALIGenericIterator(
          [pipe],
          ["data"],
          reader_name="reader",
          last_batch_policy=LastBatchPolicy.PARTIAL,
      )
  
file_path: DALI/dali/test/python/jax_plugin/test_iterator.py
imports: import numpy
  import jax
  import jax.numpy
  import jax.dlpack
  from utils import iterator_function_def
  from nvidia.dali.plugin.jax import DALIGenericIterator
  from nvidia.dali.pipeline import pipeline_def
  from nvidia.dali.plugin.base_iterator import LastBatchPolicy
  from nose_utils import raises
  import itertools
label_desc: Use image node_id 3 for calling a global function with example usage: test_iterator_last_batch_policy_partial_exception() without return types