We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 86dbe89 commit 67b9ac6Copy full SHA for 67b9ac6
1 file changed
src/xpk/parser/cluster.py
@@ -269,9 +269,13 @@ def set_cluster_create_ray_parser(cluster_create_ray_parser: ArgumentParser):
269
add_shared_cluster_create_required_arguments(
270
cluster_create_ray_required_arguments
271
)
272
- add_tpu_type_argument(cluster_create_ray_required_arguments, required=True)
+ 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)
278
- # TODO(bzmarke): Add --device-type to support GPU/CPU
279
cluster_create_ray_required_arguments.add_argument(
280
'--ray-version',
281
type=str,
0 commit comments