Skip to content

Commit

Permalink
Merge branch 'air-dataset-split-refactor' of github.com:amogkam/ray i…
Browse files Browse the repository at this point in the history
…nto air-dataset-split-refactor
  • Loading branch information
amogkam committed Apr 21, 2022
2 parents 13537ca + 7e7c927 commit 8783055
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/ray/train/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,7 @@ class _RayDatasetSpec:
can be passed in as a ``Dict`` that maps each name key to a Dataset value,
and each Dataset can be accessed from the training function by passing in a
`dataset_name` argument to ``train.get_dataset_shard()``.
dataset_split_fn: An optional callbale to specify how the provided ``dataset``
dataset_split_fn: An optional callable to specify how the provided ``dataset``
should be split across the training workers. It is expected to take in two
arguments. The first one is the ``dataset``, just as is passed in to the
``_RayDatasetSpec``. The second argument is a list of the ActorHandles of the
Expand Down

0 comments on commit 8783055

Please sign in to comment.