From 7e7c927b18523ee29e76ad56d6cf0bc4a1194b3d Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Wed, 20 Apr 2022 18:24:39 -0700 Subject: [PATCH] Update python/ray/train/utils.py Co-authored-by: Eric Liang --- python/ray/train/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/train/utils.py b/python/ray/train/utils.py index 71451714a571..aeda801b2727 100644 --- a/python/ray/train/utils.py +++ b/python/ray/train/utils.py @@ -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