hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[docs] Fix broken code example in docstring for DataParallelTrainer args

Browse source
This commit is contained in:
Eric Liang 2022-05-14 20:48:45 -07:00 • committed by GitHub
parent 684e395c5d
commit 9381dd174e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/ray/ml/train/data_parallel_trainer.py
View file

@ -196,7 +196,9 @@ class DataParallelTrainer(Trainer):
def __init__(self, train_loop_per_worker, my_backend_config:
MyBackendConfig, **kwargs):
super().__init__(train_loop_per_worker, my_backend_config, **kwargs)
super().__init__(
train_loop_per_worker,
backend_config=my_backend_config, **kwargs)
Args:
train_loop_per_worker: The training function to execute.
@ -232,8 +234,8 @@ class DataParallelTrainer(Trainer):
def __init__(
self,
*,
train_loop_per_worker: Union[Callable[[], None], Callable[[Dict], None]],
*,
train_loop_config: Optional[Dict] = None,
backend_config: Optional[BackendConfig] = None,
scaling_config: Optional[ScalingConfig] = None,