Update distributed training CLI to add "ray" framework. (#372)

oracle · Oct 16, 2023 · ca6d812 · ca6d812
2 parents 9018bde + dd2b4b3
commit ca6d812
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ads/opctl/distributed/cli.py b/ads/opctl/distributed/cli.py
@@ -30,7 +30,14 @@ def commands():
     "-f",
     help="Distributed training framework type",
     type=click.Choice(
-        ["dask", "horovod-tensorflow", "horovod-pytorch", "pytorch", "tensorflow"]
+        [
+            "dask",
+            "horovod-tensorflow",
+            "horovod-pytorch",
+            "pytorch",
+            "tensorflow",
+            "ray",
+        ]
     ),
     default=None,
     required=True,