From 9d3438a96d1ce07939358fa214bbf54fd11cc37e Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Fri, 31 Jul 2026 22:23:16 +0000 Subject: [PATCH 1/3] fix(cli): align cluster NVIDIA selector handling Co-authored-by: Blaine Kasten --- .../lib/cli/api/beta/clusters/create.py | 50 ++++++++++---- src/together/lib/cli/utils/_help_examples.py | 5 +- tests/cli/test_beta_clusters.py | 68 ++++++++++++++++--- 3 files changed, 99 insertions(+), 24 deletions(-) diff --git a/src/together/lib/cli/api/beta/clusters/create.py b/src/together/lib/cli/api/beta/clusters/create.py index 96026657c..c85ed2c85 100644 --- a/src/together/lib/cli/api/beta/clusters/create.py +++ b/src/together/lib/cli/api/beta/clusters/create.py @@ -22,11 +22,13 @@ Optional[Literal["RESERVED", "ON_DEMAND", "SCHEDULED_CAPACITY"]], Parameter(help="Billing type to use for the cluster"), ] -NvidiaDriverVersionParameter = Annotated[Optional[str], Parameter(help="Nvidia driver version to use for the cluster")] -CudaVersionParameter = Annotated[Optional[str], Parameter(help="CUDA version to use for the cluster")] +NvidiaDriverVersionParameter = Annotated[ + Optional[str], Parameter(help="Legacy NVIDIA driver selector; pair with --cuda-version") +] +CudaVersionParameter = Annotated[Optional[str], Parameter(help="Legacy CUDA selector; pair with --nvidia-driver-version")] OSParameter = Annotated[Optional[str], Parameter(help="Operating system for NVIDIA version selection")] NvidiaVersionIDParameter = Annotated[ - Optional[str], Parameter(help="NVIDIA version catalog ID to use directly for the cluster") + Optional[str], Parameter(help="Canonical NVIDIA version catalog ID to use for the cluster") ] DurationDaysParameter = Annotated[ Optional[int], Parameter(help="Duration in days to keep the cluster running for reserved clusters") @@ -151,24 +153,46 @@ async def _set_nvidia_version_params( os_name: str | None, ) -> None: semantic_version_given = any(value is not None for value in (nvidia_driver_version, cuda_version, os_name)) - if nvidia_version_id and semantic_version_given: - raise TogetherError("Use either --nvidia-version-id or --nvidia-driver-version/--cuda-version/--os, not both.") - has_driver = nvidia_driver_version is not None has_cuda = cuda_version is not None - if not nvidia_version_id and (has_driver != has_cuda or (os_name is not None and not has_driver)): + if has_driver != has_cuda or (os_name is not None and not has_driver): raise TogetherError("--nvidia-driver-version and --cuda-version must be provided together; --os requires both.") if nvidia_version_id: params["nvidia_version_id"] = nvidia_version_id - params.pop("nvidia_driver_version", None) - params.pop("cuda_version", None) - return + if not semantic_version_given: + params.pop("nvidia_driver_version", None) + params.pop("cuda_version", None) + return - if not interactive and not has_driver: - raise TogetherError( - "Use --nvidia-version-id or provide --nvidia-driver-version and --cuda-version in non-interactive mode." + if os_name is None: + return + + region = params.get("region") + if not region: + raise TogetherError("--region is required when selecting an NVIDIA version.") + + if catalog is None: + catalog = await config.client.beta.clusters.list_regions() + selected = _resolve_nvidia_version( + catalog, + region=region, + nvidia_driver_version=nvidia_driver_version, + cuda_version=cuda_version, + os_name=os_name, ) + if selected.id and selected.id != nvidia_version_id: + raise TogetherError( + f"--nvidia-version-id {nvidia_version_id!r} does not match " + f"--nvidia-driver-version/--cuda-version/--os selection {selected.id!r}." + ) + return + + if not has_driver: + if not interactive: + params.pop("nvidia_driver_version", None) + params.pop("cuda_version", None) + return if has_driver and os_name is None: return diff --git a/src/together/lib/cli/utils/_help_examples.py b/src/together/lib/cli/utils/_help_examples.py index f5fe1715b..92b5635c7 100644 --- a/src/together/lib/cli/utils/_help_examples.py +++ b/src/together/lib/cli/utils/_help_examples.py @@ -520,7 +520,7 @@ [primary]tg beta clusters create --non-interactive \\ --name my-cluster --cluster-type KUBERNETES --gpu-type H100_SXM \\ --region us-central-8 --num-gpus 8 --billing-type ON_DEMAND \\ - --nvidia-driver-version 565 --cuda-version 12.6 --volume [/primary] + --nvidia-version-id nvidia-565-22 --volume [/primary] [dim]-[/dim] Update or delete a cluster: [primary]tg beta clusters update --num-gpus 16 --cluster-type KUBERNETES[/primary] @@ -546,8 +546,7 @@ --region us-central-8 \\ --num-gpus 8 \\ --billing-type ON_DEMAND \\ - --nvidia-driver-version 565 \\ - --cuda-version 12.6 \\ + --nvidia-version-id nvidia-565-22 \\ --volume [/primary] """ diff --git a/tests/cli/test_beta_clusters.py b/tests/cli/test_beta_clusters.py index f9126009f..3dab90ad0 100644 --- a/tests/cli/test_beta_clusters.py +++ b/tests/cli/test_beta_clusters.py @@ -666,10 +666,9 @@ class TestBetaClustersNvidiaVersionSelection: [ ("595", None, None, "must be provided together"), (None, None, "ubuntu-24.04", "--os requires both"), - (None, None, None, "Use --nvidia-version-id"), ], ) - async def test_non_interactive_selection_requires_complete_selector( + async def test_non_interactive_selection_rejects_incomplete_selector( self, nvidia_driver_version: str | None, cuda_version: str | None, @@ -688,6 +687,25 @@ async def test_non_interactive_selection_requires_complete_selector( os_name=os_name, ) + @pytest.mark.asyncio + async def test_non_interactive_selection_can_omit_selector(self) -> None: + params: dict[str, Any] = {} + + await create_cli._set_nvidia_version_params( + config=cast(Any, None), + params=params, + catalog=None, + interactive=False, + nvidia_version_id=None, + nvidia_driver_version=None, + cuda_version=None, + os_name=None, + ) + + assert "nvidia_version_id" not in params + assert "nvidia_driver_version" not in params + assert "cuda_version" not in params + @pytest.mark.asyncio async def test_interactive_explicit_legacy_pair_passes_through(self) -> None: params: dict[str, Any] = { @@ -861,7 +879,7 @@ def test_retrieve_json(self, respx_mock: MockRouter, cli_runner: CliRunner) -> N class TestBetaClustersCreate: - def test_invalid_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRunner) -> None: + def test_incomplete_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRunner) -> None: result = cli_runner.invoke( [ "beta", @@ -872,12 +890,8 @@ def test_invalid_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRunne "KUBERNETES", "--gpu-type", "H100_SXM", - "--nvidia-version-id", - "nvidia-595-24", "--nvidia-driver-version", "595", - "--cuda-version", - "13.2", "--region", "us-central-8", "--num-gpus", @@ -891,9 +905,42 @@ def test_invalid_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRunne assert result.exit_code == 1 assert json.loads(result.output) == { - "error": "Use either --nvidia-version-id or --nvidia-driver-version/--cuda-version/--os, not both." + "error": "--nvidia-driver-version and --cuda-version must be provided together; --os requires both." } + @pytest.mark.respx(base_url=base_url) + def test_create_non_interactive_can_omit_nvidia_selector( + self, respx_mock: MockRouter, cli_runner: CliRunner + ) -> None: + created = _cluster_body("new-id", "default-selector") + route = respx_mock.post("/compute/clusters").mock(return_value=httpx.Response(200, json=created)) + result = cli_runner.invoke( + [ + "beta", + "clusters", + "create", + "--non-interactive", + "--cluster-type", + "KUBERNETES", + "--gpu-type", + "H100_SXM", + "--region", + "us-central-8", + "--num-gpus", + "8", + "--billing-type", + "ON_DEMAND", + "--name", + "default-selector", + ], + ) + + assert result.exit_code == 0, result.output + body = json.loads(cast(Call, route.calls[0]).request.content.decode()) + assert "nvidia_version_id" not in body + assert "nvidia_driver_version" not in body + assert "cuda_version" not in body + @pytest.mark.respx(base_url=base_url) def test_create_non_interactive_posts_expected_body(self, respx_mock: MockRouter, cli_runner: CliRunner) -> None: created = _cluster_body("new-id", "together-py-testing-suite") @@ -1028,6 +1075,8 @@ def test_create_accepts_new_cluster_params(self, respx_mock: MockRouter, cli_run "565", "--cuda-version", "12.6", + "--nvidia-version-id", + "nvidia-565-22", "--region", "us-central-8", "--num-gpus", @@ -1091,6 +1140,9 @@ def test_create_accepts_new_cluster_params(self, respx_mock: MockRouter, cli_run assert body["reservation_end_time"] == "2026-06-02T00:00:00Z" assert body["slurm_image"] == "slurm:latest" assert body["slurm_shm_size_gib"] == 32 + assert body["nvidia_driver_version"] == "565" + assert body["cuda_version"] == "12.6" + assert body["nvidia_version_id"] == "nvidia-565-22" assert result.exit_code == 0 From 36dab9916071a77d139985b90fffdabe570f686c Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Wed, 12 Aug 2026 18:44:17 +0000 Subject: [PATCH 2/3] fix(cli): rename cluster create --nvidia-driver-version to --driver Use the shorter --driver flag for the legacy NVIDIA driver selector in beta cluster create, and update help/error strings and CLI tests. Co-authored-by: Blaine Kasten --- src/together/lib/cli/api/beta/clusters/create.py | 16 +++++++--------- tests/cli/test_beta_clusters.py | 10 +++++----- tests/cli/test_json_mode_pipeable_to_jq.py | 2 +- 3 files changed, 13 insertions(+), 15 deletions(-) diff --git a/src/together/lib/cli/api/beta/clusters/create.py b/src/together/lib/cli/api/beta/clusters/create.py index c85ed2c85..faff97fd4 100644 --- a/src/together/lib/cli/api/beta/clusters/create.py +++ b/src/together/lib/cli/api/beta/clusters/create.py @@ -22,10 +22,8 @@ Optional[Literal["RESERVED", "ON_DEMAND", "SCHEDULED_CAPACITY"]], Parameter(help="Billing type to use for the cluster"), ] -NvidiaDriverVersionParameter = Annotated[ - Optional[str], Parameter(help="Legacy NVIDIA driver selector; pair with --cuda-version") -] -CudaVersionParameter = Annotated[Optional[str], Parameter(help="Legacy CUDA selector; pair with --nvidia-driver-version")] +DriverParameter = Annotated[Optional[str], Parameter(help="Legacy NVIDIA driver selector; pair with --cuda-version")] +CudaVersionParameter = Annotated[Optional[str], Parameter(help="Legacy CUDA selector; pair with --driver")] OSParameter = Annotated[Optional[str], Parameter(help="Operating system for NVIDIA version selection")] NvidiaVersionIDParameter = Annotated[ Optional[str], Parameter(help="Canonical NVIDIA version catalog ID to use for the cluster") @@ -156,7 +154,7 @@ async def _set_nvidia_version_params( has_driver = nvidia_driver_version is not None has_cuda = cuda_version is not None if has_driver != has_cuda or (os_name is not None and not has_driver): - raise TogetherError("--nvidia-driver-version and --cuda-version must be provided together; --os requires both.") + raise TogetherError("--driver and --cuda-version must be provided together; --os requires both.") if nvidia_version_id: params["nvidia_version_id"] = nvidia_version_id @@ -184,7 +182,7 @@ async def _set_nvidia_version_params( if selected.id and selected.id != nvidia_version_id: raise TogetherError( f"--nvidia-version-id {nvidia_version_id!r} does not match " - f"--nvidia-driver-version/--cuda-version/--os selection {selected.id!r}." + f"--driver/--cuda-version/--os selection {selected.id!r}." ) return @@ -228,7 +226,7 @@ async def create( num_gpus: NumGpusParameter = None, region: RegionParameter = None, billing_type: BillingTypeParameter = None, - nvidia_driver_version: NvidiaDriverVersionParameter = None, + driver: DriverParameter = None, cuda_version: CudaVersionParameter = None, os: OSParameter = None, nvidia_version_id: NvidiaVersionIDParameter = None, @@ -258,7 +256,7 @@ async def create( num_gpus=num_gpus, region=region, billing_type=billing_type, - nvidia_driver_version=nvidia_driver_version, + nvidia_driver_version=driver, cuda_version=cuda_version, duration_days=duration_days, gpu_type=gpu_type, @@ -342,7 +340,7 @@ async def create( catalog=catalog, interactive=interactive, nvidia_version_id=nvidia_version_id, - nvidia_driver_version=nvidia_driver_version, + nvidia_driver_version=driver, cuda_version=cuda_version, os_name=os, ) diff --git a/tests/cli/test_beta_clusters.py b/tests/cli/test_beta_clusters.py index 3dab90ad0..e49c25160 100644 --- a/tests/cli/test_beta_clusters.py +++ b/tests/cli/test_beta_clusters.py @@ -890,7 +890,7 @@ def test_incomplete_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRu "KUBERNETES", "--gpu-type", "H100_SXM", - "--nvidia-driver-version", + "--driver", "595", "--region", "us-central-8", @@ -905,7 +905,7 @@ def test_incomplete_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRu assert result.exit_code == 1 assert json.loads(result.output) == { - "error": "--nvidia-driver-version and --cuda-version must be provided together; --os requires both." + "error": "--driver and --cuda-version must be provided together; --os requires both." } @pytest.mark.respx(base_url=base_url) @@ -955,7 +955,7 @@ def test_create_non_interactive_posts_expected_body(self, respx_mock: MockRouter "KUBERNETES", "--gpu-type", "H100_SXM", - "--nvidia-driver-version", + "--driver", "565", "--cuda-version", "12.6", @@ -1034,7 +1034,7 @@ def test_create_semantic_nvidia_selection_posts_resolved_id( "KUBERNETES", "--gpu-type", "H100_SXM", - "--nvidia-driver-version", + "--driver", "595", "--cuda-version", "13.2", @@ -1071,7 +1071,7 @@ def test_create_accepts_new_cluster_params(self, respx_mock: MockRouter, cli_run "SLURM", "--gpu-type", "H100_SXM", - "--nvidia-driver-version", + "--driver", "565", "--cuda-version", "12.6", diff --git a/tests/cli/test_json_mode_pipeable_to_jq.py b/tests/cli/test_json_mode_pipeable_to_jq.py index e88481d6b..86c2b7949 100644 --- a/tests/cli/test_json_mode_pipeable_to_jq.py +++ b/tests/cli/test_json_mode_pipeable_to_jq.py @@ -140,7 +140,7 @@ def test_beta_clusters_json_mode(self) -> None: beta_clusters = JSONValidator(("beta", "clusters")) beta_clusters.run_and_assert( "create --non-interactive --cluster-type KUBERNETES --gpu-type H100_SXM " - "--nvidia-driver-version 565 --cuda-version 12.6 --region us-central-8 --num-gpus 8 " + "--driver 565 --cuda-version 12.6 --region us-central-8 --num-gpus 8 " "--billing-type ON_DEMAND --name together-py-testing-suite --volume 123" ) beta_clusters.run_and_assert("delete cluster-123") From f2d148da49481dbadac04534bc64c52b29d7724b Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Wed, 12 Aug 2026 18:55:09 +0000 Subject: [PATCH 3/3] fix(cli): expose NVIDIA version ID as --driver Revert the earlier rename of the legacy selector. Keep --nvidia-driver-version/--cuda-version for semantic pairs, and rename the canonical NVIDIA version catalog flag from --nvidia-version-id to --driver. Co-authored-by: Blaine Kasten --- .../lib/cli/api/beta/clusters/create.py | 30 +++++++++++-------- src/together/lib/cli/utils/_help_examples.py | 4 +-- tests/cli/test_beta_clusters.py | 18 +++++------ tests/cli/test_json_mode_pipeable_to_jq.py | 2 +- 4 files changed, 30 insertions(+), 24 deletions(-) diff --git a/src/together/lib/cli/api/beta/clusters/create.py b/src/together/lib/cli/api/beta/clusters/create.py index faff97fd4..a950ad6e0 100644 --- a/src/together/lib/cli/api/beta/clusters/create.py +++ b/src/together/lib/cli/api/beta/clusters/create.py @@ -22,10 +22,14 @@ Optional[Literal["RESERVED", "ON_DEMAND", "SCHEDULED_CAPACITY"]], Parameter(help="Billing type to use for the cluster"), ] -DriverParameter = Annotated[Optional[str], Parameter(help="Legacy NVIDIA driver selector; pair with --cuda-version")] -CudaVersionParameter = Annotated[Optional[str], Parameter(help="Legacy CUDA selector; pair with --driver")] +NvidiaDriverVersionParameter = Annotated[ + Optional[str], Parameter(help="Legacy NVIDIA driver selector; pair with --cuda-version") +] +CudaVersionParameter = Annotated[ + Optional[str], Parameter(help="Legacy CUDA selector; pair with --nvidia-driver-version") +] OSParameter = Annotated[Optional[str], Parameter(help="Operating system for NVIDIA version selection")] -NvidiaVersionIDParameter = Annotated[ +DriverParameter = Annotated[ Optional[str], Parameter(help="Canonical NVIDIA version catalog ID to use for the cluster") ] DurationDaysParameter = Annotated[ @@ -112,7 +116,7 @@ def _resolve_nvidia_version( ) if len(matches) > 1: choices = "; ".join(_format_nvidia_version(version) for version in matches) - guidance = "Use --nvidia-version-id." if os_name else "Add --os or use --nvidia-version-id." + guidance = "Use --driver." if os_name else "Add --os or use --driver." raise TogetherError( f"Multiple NVIDIA versions match {requested} in region '{region}'. {guidance} Matches: {choices}" ) @@ -154,7 +158,9 @@ async def _set_nvidia_version_params( has_driver = nvidia_driver_version is not None has_cuda = cuda_version is not None if has_driver != has_cuda or (os_name is not None and not has_driver): - raise TogetherError("--driver and --cuda-version must be provided together; --os requires both.") + raise TogetherError( + "--nvidia-driver-version and --cuda-version must be provided together; --os requires both." + ) if nvidia_version_id: params["nvidia_version_id"] = nvidia_version_id @@ -181,8 +187,8 @@ async def _set_nvidia_version_params( ) if selected.id and selected.id != nvidia_version_id: raise TogetherError( - f"--nvidia-version-id {nvidia_version_id!r} does not match " - f"--driver/--cuda-version/--os selection {selected.id!r}." + f"--driver {nvidia_version_id!r} does not match " + f"--nvidia-driver-version/--cuda-version/--os selection {selected.id!r}." ) return @@ -226,10 +232,10 @@ async def create( num_gpus: NumGpusParameter = None, region: RegionParameter = None, billing_type: BillingTypeParameter = None, - driver: DriverParameter = None, + nvidia_driver_version: NvidiaDriverVersionParameter = None, cuda_version: CudaVersionParameter = None, os: OSParameter = None, - nvidia_version_id: NvidiaVersionIDParameter = None, + driver: DriverParameter = None, duration_days: DurationDaysParameter = None, gpu_type: GpuTypeParameter = None, cluster_type: ClusterTypeParameter = None, @@ -256,7 +262,7 @@ async def create( num_gpus=num_gpus, region=region, billing_type=billing_type, - nvidia_driver_version=driver, + nvidia_driver_version=nvidia_driver_version, cuda_version=cuda_version, duration_days=duration_days, gpu_type=gpu_type, @@ -339,8 +345,8 @@ async def create( params=params, catalog=catalog, interactive=interactive, - nvidia_version_id=nvidia_version_id, - nvidia_driver_version=driver, + nvidia_version_id=driver, + nvidia_driver_version=nvidia_driver_version, cuda_version=cuda_version, os_name=os, ) diff --git a/src/together/lib/cli/utils/_help_examples.py b/src/together/lib/cli/utils/_help_examples.py index 92b5635c7..1b44fb531 100644 --- a/src/together/lib/cli/utils/_help_examples.py +++ b/src/together/lib/cli/utils/_help_examples.py @@ -520,7 +520,7 @@ [primary]tg beta clusters create --non-interactive \\ --name my-cluster --cluster-type KUBERNETES --gpu-type H100_SXM \\ --region us-central-8 --num-gpus 8 --billing-type ON_DEMAND \\ - --nvidia-version-id nvidia-565-22 --volume [/primary] + --driver nvidia-565-22 --volume [/primary] [dim]-[/dim] Update or delete a cluster: [primary]tg beta clusters update --num-gpus 16 --cluster-type KUBERNETES[/primary] @@ -546,7 +546,7 @@ --region us-central-8 \\ --num-gpus 8 \\ --billing-type ON_DEMAND \\ - --nvidia-version-id nvidia-565-22 \\ + --driver nvidia-565-22 \\ --volume [/primary] """ diff --git a/tests/cli/test_beta_clusters.py b/tests/cli/test_beta_clusters.py index e49c25160..a9d3028b2 100644 --- a/tests/cli/test_beta_clusters.py +++ b/tests/cli/test_beta_clusters.py @@ -801,7 +801,7 @@ def test_semantic_selection_uses_os_to_disambiguate_duplicate_cuda_rows(self) -> assert selected.id == "nvidia-595-22" def test_semantic_selection_requires_disambiguation_for_duplicate_cuda_rows(self) -> None: - with pytest.raises(TogetherError, match="Add --os or use --nvidia-version-id"): + with pytest.raises(TogetherError, match="Add --os or use --driver"): create_cli._resolve_nvidia_version( ClusterListRegionsResponse(**_REGIONS_BODY), region="us-central-8", @@ -839,7 +839,7 @@ def test_semantic_selection_with_duplicate_os_recommends_id_only(self) -> None: ) ) - with pytest.raises(TogetherError, match=r"Use --nvidia-version-id\. Matches"): + with pytest.raises(TogetherError, match=r"Use --driver\. Matches"): create_cli._resolve_nvidia_version( catalog, region="us-central-8", @@ -890,7 +890,7 @@ def test_incomplete_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRu "KUBERNETES", "--gpu-type", "H100_SXM", - "--driver", + "--nvidia-driver-version", "595", "--region", "us-central-8", @@ -905,7 +905,7 @@ def test_incomplete_nvidia_selector_is_json_in_json_mode(self, cli_runner: CliRu assert result.exit_code == 1 assert json.loads(result.output) == { - "error": "--driver and --cuda-version must be provided together; --os requires both." + "error": "--nvidia-driver-version and --cuda-version must be provided together; --os requires both." } @pytest.mark.respx(base_url=base_url) @@ -955,7 +955,7 @@ def test_create_non_interactive_posts_expected_body(self, respx_mock: MockRouter "KUBERNETES", "--gpu-type", "H100_SXM", - "--driver", + "--nvidia-driver-version", "565", "--cuda-version", "12.6", @@ -997,7 +997,7 @@ def test_create_direct_nvidia_version_id_posts_id(self, respx_mock: MockRouter, "KUBERNETES", "--gpu-type", "H100_SXM", - "--nvidia-version-id", + "--driver", "nvidia-595-24", "--region", "us-central-8", @@ -1034,7 +1034,7 @@ def test_create_semantic_nvidia_selection_posts_resolved_id( "KUBERNETES", "--gpu-type", "H100_SXM", - "--driver", + "--nvidia-driver-version", "595", "--cuda-version", "13.2", @@ -1071,11 +1071,11 @@ def test_create_accepts_new_cluster_params(self, respx_mock: MockRouter, cli_run "SLURM", "--gpu-type", "H100_SXM", - "--driver", + "--nvidia-driver-version", "565", "--cuda-version", "12.6", - "--nvidia-version-id", + "--driver", "nvidia-565-22", "--region", "us-central-8", diff --git a/tests/cli/test_json_mode_pipeable_to_jq.py b/tests/cli/test_json_mode_pipeable_to_jq.py index 86c2b7949..e88481d6b 100644 --- a/tests/cli/test_json_mode_pipeable_to_jq.py +++ b/tests/cli/test_json_mode_pipeable_to_jq.py @@ -140,7 +140,7 @@ def test_beta_clusters_json_mode(self) -> None: beta_clusters = JSONValidator(("beta", "clusters")) beta_clusters.run_and_assert( "create --non-interactive --cluster-type KUBERNETES --gpu-type H100_SXM " - "--driver 565 --cuda-version 12.6 --region us-central-8 --num-gpus 8 " + "--nvidia-driver-version 565 --cuda-version 12.6 --region us-central-8 --num-gpus 8 " "--billing-type ON_DEMAND --name together-py-testing-suite --volume 123" ) beta_clusters.run_and_assert("delete cluster-123")