Skip to content

Commit

Permalink
Fix data_iterator docs (#5140)
Browse files Browse the repository at this point in the history
Signed-off-by: Albert Wolant <awolant@nvidia.com>
  • Loading branch information
awolant committed Nov 3, 2023
1 parent 2bc5dbf commit abdb213
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions dali/python/nvidia/dali/plugin/jax/iterator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit abdb213

Please sign in to comment.