code: def test_positional_sharding_workflow_with_iterator():
      mesh = mesh_utils.create_device_mesh((jax.device_count(), 1))
      sharding = PositionalSharding(mesh)
  
      run_sharding_iterator_test(sharding)
  
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 8 for calling a global function with example usage: test_positional_sharding_workflow_with_iterator() without return types