Skip to content

Commit ca6d812

Browse files
authored
Update distributed training CLI to add "ray" framework. (#372)
2 parents 9018bde + dd2b4b3 commit ca6d812

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

ads/opctl/distributed/cli.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,14 @@ def commands():
3030
"-f",
3131
help="Distributed training framework type",
3232
type=click.Choice(
33-
["dask", "horovod-tensorflow", "horovod-pytorch", "pytorch", "tensorflow"]
33+
[
34+
"dask",
35+
"horovod-tensorflow",
36+
"horovod-pytorch",
37+
"pytorch",
38+
"tensorflow",
39+
"ray",
40+
]
3441
),
3542
default=None,
3643
required=True,

0 commit comments

Comments
 (0)