Fix DALIGenericPeekableIterator missing pmap_compatible parameter.

- The pmap_compatible argument was added to DALIGenericIterator in
  ff8655b but was not propagated to DALIGenericPeekableIterator,
  causing test_iterators_init_method_api_compatibility to fail.

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>

Author: JanuszL

dali/python/nvidia/dali/plugin/jax/clu.py

@@ -101,6 +101,14 @@ class DALIGenericPeekableIterator(DALIGenericIterator):
                 `jax.sharding.Sharding` compatible object that, if present, will be used to
                 build an output jax.Array for each category. If ``None``, the iterator returns
                 values compatible with pmapped JAX functions, if multiple pipelines are provided.
+    pmap_compatible : bool, optional, default = None
+                Controls whether the iterator produces outputs with a leading device axis
+                compatible with ``jax.pmap``. When ``None`` (default), it is inferred
+                automatically: ``True`` when ``devices`` is provided, ``False`` otherwise.
+                Set to ``True`` explicitly to force pmap-compatible output (shape
+                ``[num_devices, batch_per_device, ...]``) without using the ``devices``
+                argument. Set to ``False`` to suppress the device axis even when ``devices``
+                is provided.
 
     Example
     -------
@@ -133,6 +141,7 @@ def __init__(
         last_batch_policy: LastBatchPolicy = LastBatchPolicy.FILL,
         prepare_first_batch: bool = True,
         sharding: Optional[Sharding] = None,
+        pmap_compatible: Optional[bool] = None,
     ):
         super().__init__(
             pipelines,
@@ -144,6 +153,7 @@ def __init__(
             last_batch_policy,
             prepare_first_batch,
             sharding,
+            pmap_compatible,
         )
         self._mutex = threading.Lock()
         self._pool = None