Skip to content

Commit 67b9ac6

Browse files
authored
feat: support --device-type for xpk cluster create-ray (#1175)
1 parent 86dbe89 commit 67b9ac6

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

src/xpk/parser/cluster.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -269,9 +269,13 @@ def set_cluster_create_ray_parser(cluster_create_ray_parser: ArgumentParser):
269269
add_shared_cluster_create_required_arguments(
270270
cluster_create_ray_required_arguments
271271
)
272-
add_tpu_type_argument(cluster_create_ray_required_arguments, required=True)
272+
cluster_device_group = (
273+
cluster_create_ray_required_arguments.add_mutually_exclusive_group(
274+
required=True
275+
)
276+
)
277+
add_tpu_and_device_type_arguments(cluster_device_group)
273278

274-
# TODO(bzmarke): Add --device-type to support GPU/CPU
275279
cluster_create_ray_required_arguments.add_argument(
276280
'--ray-version',
277281
type=str,

0 commit comments

Comments
 (0)