From 40fdb7630749dbd2ea5d8d5cff24d3316418eb16 Mon Sep 17 00:00:00 2001 From: Albert Wolant Date: Thu, 2 Nov 2023 14:04:22 +0100 Subject: [PATCH] Fix docs 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,