Skip to content

air: show compute options in accelerator type help - #7275

Merged
caroline-db merged 4 commits into
mainfrom
air/accelerator-type-help
Oct 31, 2026
Merged

air: show compute options in accelerator type help#7275
caroline-db merged 4 commits into
mainfrom
air/accelerator-type-help

Conversation

@caroline-db

@caroline-db caroline-db commented Oct 30, 2026

Copy link
Copy Markdown
Contributor

Changes

  • Show workspace-supported accelerator types in databricks air run -h config.compute.accelerator_type.
  • Reuse the existing workspace lookup and fallback behavior from config.compute.

Why

Users naturally inspect the accelerator_type field for supported values, but the dynamic list was only shown on its parent object.

Tests

  • ./task fmt
  • ./task checks
  • ./task lint
  • go test ./cmd/air
  • go test ./acceptance -run TestAccept/air/config-help -tail
  • Verified the locally built CLI against a live workspace.

This PR was written by Codex.

@caroline-db
caroline-db requested review from a team as code owners October 30, 2026 19:01
@caroline-db
caroline-db requested a review from anton-107 October 30, 2026 19:01
@github-actions github-actions Bot added the AIR Databricks AI Runtime CLI label Oct 30, 2026
@github-actions

github-actions Bot commented Oct 30, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

Commit: 5fc3b1871 | Workflow run

Regression test report

Selected 1 changed tests (limit=20, 0 not selected)

1 tests: regression=1 unreleased=0 coverage=0 failing=0

test branch main (343ae60) latest (v1.23.0)
TestAccept/air/config-help ✅ ❌ ❌
TestAccept/air/config-help: branch ✅ | main (343ae60) ❌ | latest (v1.23.0) ❌

main (343ae60):

acceptance_test.go:1341: Diff:
        --- air/config-help/output.txt
        +++ /tmp/TestAcceptairconfig-help2836127909/002/output.txt
        @@ -128,11 +128,6 @@
         config.compute.accelerator_type
           Which accelerator to run on, e.g. GPU_1xA10. See https://docs.databricks.com/aws/en/machine-learning/ai-runtime/cli/yaml-config#reference for the current list of supported GPU types. Matched case-sensitively.
         
        -  Supported accelerator types:
        -    GPU_1xH100 - 1 accelerator per node
        -    GPU_8xH100 - 8 accelerators per node
        -    GPU_1xRTX6000 - 1 accelerator per node
        -
           Type:     string
           Required: no
         
        
FAIL

latest (v1.23.0): same output as main (343ae60)

@eng-dev-ecosystem-bot

eng-dev-ecosystem-bot commented Oct 30, 2026 •
edited
Loading

Copy link
Copy Markdown
Collaborator

Integration test report

Commit: 5fc3b18

Run: 41139038113

Env 🔄​flaky ✅​pass 🙈​skip Time
✅​ aws linux-2core-8gb 281 16 5:13
✅​ aws-windows-latest-4core-16gb 283 14 3:33
✅​ azure linux-2core-8gb 280 16 5:15
🔄​ azure-windows-latest-4core-16gb 2 280 14 3:49
✅​ gcp linux-2core-8gb 281 16 5:10
✅​ gcp-windows-latest-4core-16gb 283 14 3:35
Test Name azure-windows-latest-4core-16gb
🔄​ TestFilerRecursiveDelete 🔄​f
🔄​ TestFilerRecursiveDelete/files 🔄​f
Top 6 slowest tests (at least 2 minutes):
duration env testname
3:50 gcp linux-2core-8gb TestAccept
3:49 azure linux-2core-8gb TestAccept
3:48 aws linux-2core-8gb TestAccept
3:28 gcp-windows-latest-4core-16gb TestAccept
3:26 azure-windows-latest-4core-16gb TestAccept
3:26 aws-windows-latest-4core-16gb TestAccept

@caroline-db
caroline-db force-pushed the air/accelerator-type-help branch from 8a249d3 to 6573cdc Compare October 30, 2026 20:06
maggiewang-db
maggiewang-db reviewed Oct 30, 2026
View reviewed changes
Comment thread acceptance/air/config-help/output.txt Resolved
maggiewang-db
maggiewang-db approved these changes Oct 30, 2026
View reviewed changes
Comment thread acceptance/air/config-help/output.txt Resolved
@caroline-db
caroline-db added this pull request to the merge queue Oct 31, 2026
Merged via the queue into main with commit 3aa5c96 Oct 31, 2026
28 checks passed
@caroline-db
caroline-db deleted the air/accelerator-type-help branch October 31, 2026 01:45
@eng-dev-ecosystem-bot

eng-dev-ecosystem-bot commented Oct 31, 2026

Copy link
Copy Markdown
Collaborator

Integration test report

Commit: 3aa5c96

Run: 41156943625

Env 🔄​flaky ✅​pass 🙈​skip Time
🔄​ aws linux-2core-8gb 3 1501 1226 125:54
🔄​ azure linux-2core-8gb 3 1348 1278 109:22
✅​ gcp linux-2core-8gb 1338 1283 114:28
Test Name aws linux-2core-8gb azure linux-2core-8gb
🔄​ TestAccept 🔄​f 🔄​f
🔄​ TestAccept/bundle/resources/apps/lifecycle-started ✅​p 🔄​f
🔄​ TestAccept/bundle/resources/apps/lifecycle-started/DMS=true ✅​p 🔄​f
🔄​ TestAccept/bundle/resources/postgres_databases/replace_existing 🔄​f 🙈​s
🔄​ TestAccept/bundle/resources/postgres_databases/replace_existing/DMS=true 🔄​f
Top 50 slowest tests (at least 2 minutes):
duration env testname
10:02 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=
9:56 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=true
9:37 aws linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=true
9:14 azure linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=true
8:27 aws linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=
8:08 azure linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started/DMS=
7:12 aws linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=true
7:07 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/deploy/local_ssd_count/DMS=
7:05 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/deploy/local_ssd_count/DMS=true
7:05 aws linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=
7:04 gcp linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=true
7:02 gcp linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=
7:00 azure linux-2core-8gb TestAccept/bundle/resources/clusters/deploy/simple/DMS=true
6:55 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=
6:52 aws linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=
6:32 azure linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=
6:26 aws linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=true
6:23 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=true
6:19 aws linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=
6:19 azure linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=
6:13 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=true
6:06 aws linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=true
6:01 azure linux-2core-8gb TestAccept/bundle/config-remote-sync/multiple_resources/DMS=true
5:32 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=
5:21 azure linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=true
5:12 azure linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started/DMS=true
5:08 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=
5:08 aws linux-2core-8gb TestAccept/bundle/deploy/spark-jar-task/DMS=true
5:07 gcp linux-2core-8gb TestAccept/bundle/deploy/spark-jar-task/DMS=
4:55 aws linux-2core-8gb TestAccept/bundle/deploy/spark-jar-task/DMS=
4:41 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=true
4:05 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-omitted/DMS=
4:04 azure linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=
3:55 gcp linux-2core-8gb TestAccept/bundle/resources/apps/inline_config/DMS=true
3:55 gcp linux-2core-8gb TestAccept/bundle/resources/apps/inline_config/DMS=
3:48 gcp linux-2core-8gb TestAccept
3:44 gcp linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-omitted/DMS=true
3:43 aws linux-2core-8gb TestAccept/bundle/resources/clusters/deploy/update-after-create/DMS=
3:37 azure linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=true
3:36 azure linux-2core-8gb TestAccept/bundle/resources/clusters/lifecycle-started-toggle/DMS=
3:31 aws linux-2core-8gb TestAccept/bundle/resources/clusters/deploy/data_security_mode/DMS=true
3:26 gcp linux-2core-8gb TestAccept/bundle/invariant/no_drift/DMS=true/INPUT_CONFIG=cluster_libraries.yml.tmpl/READPLAN=1
3:25 gcp linux-2core-8gb TestAccept/bundle/invariant/no_drift/DMS=true/INPUT_CONFIG=cluster_libraries.yml.tmpl/READPLAN=
3:21 aws linux-2core-8gb TestAccept/bundle/invariant/no_drift/DMS=true/INPUT_CONFIG=cluster_libraries.yml.tmpl/READPLAN=
3:19 gcp linux-2core-8gb TestAccept/bundle/deploy/spark-jar-task/DMS=true
3:16 aws linux-2core-8gb TestAccept/bundle/resources/clusters/resize-terminated-fallback/DMS=true
3:07 gcp linux-2core-8gb TestAccept/bundle/resources/clusters/resize-terminated-fallback/DMS=true
3:07 aws linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=
3:06 aws linux-2core-8gb TestAccept/bundle/resources/apps/lifecycle-started-toggle/DMS=true
3:05 aws linux-2core-8gb TestAccept/bundle/resources/clusters/resize-terminated-fallback/DMS=

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@maggiewang-db maggiewang-db maggiewang-db approved these changes
@anton-107 anton-107 Awaiting requested review from anton-107 anton-107 is a code owner automatically assigned from databricks/eng-deco-cli

Assignees

No one assigned

Labels

AIR Databricks AI Runtime CLI

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@caroline-db @eng-dev-ecosystem-bot @maggiewang-db