From abdb213226aa85a9b6c9c727ac4360fdbc63b5b6 Mon Sep 17 00:00:00 2001 From: Albert Wolant <44801854+awolant@users.noreply.github.com> Date: Fri, 3 Nov 2023 07:57:07 +0100 Subject: [PATCH] Fix data_iterator docs (#5140) Signed-off-by: Albert Wolant --- dali/python/nvidia/dali/plugin/jax/iterator.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dali/python/nvidia/dali/plugin/jax/iterator.py b/dali/python/nvidia/dali/plugin/jax/iterator.py index 4bd2e7f555..6d56619178 100644 --- a/dali/python/nvidia/dali/plugin/jax/iterator.py +++ b/dali/python/nvidia/dali/plugin/jax/iterator.py @@ -320,9 +320,7 @@ def data_iterator( Decorated function should return DALI pipeline definition function. Decorator accepts all arguments of :meth:`nvidia.dali.plugin.base_iterator.DALIGenericIterator.__init__` and - passes them to the iterator constructor. It also accepts all arguments of - :meth:`nvidia.dali.pipeline.pipeline_def` and passes them to the pipeline definition - function. + passes them to the iterator constructor. If no `device_id` argument is passed to the decorated function, it is assumed that the first device is the one we want to use and `device_id` is set to 0. If the same argument is passed to the decorator and the decorated function,