code: def test_dali_sequential_iterator_decorator_non_default_device():
      # given
      @data_iterator(output_map=["data"], reader_name="reader")
      def iterator_function():
          return iterator_function_def()
  
      # when
      iter = iterator_function(
          num_threads=4, device_id=1, batch_size=batch_size
      )
  
      batch = next(iter)
  
      # then
      assert (
          batch["data"].device_buffers[0].device() == jax.devices()[1]
      )
  
imports: import jax
  import numpy
  import nvidia.dali.plugin.jax
  from nvidia.dali import pipeline_def
  import nvidia.dali.fn
  import nvidia.dali.types
  from nvidia.dali.plugin.jax import DALIGenericIterator, data_iterator
  from nvidia.dali.plugin.base_iterator import LastBatchPolicy
  from jax.sharding import PositionalSharding, NamedSharding, PartitionSpec, Mesh
  from jax.experimental import mesh_utils
  from utils import get_dali_tensor_gpu, iterator_function_def
  import jax.numpy
  import itertools
  from nose_utils import raises
label_desc: Use image node_id 13 for calling a global function with example usage: test_dali_sequential_iterator_decorator_non_default_device() and returns: iterator_function_def