Skip to content

[air] Remove register-image command and docker_image run support - #6657

Open
maggiewang-db wants to merge 1 commit into
mainfrom
air-remove-docker-image
Open

[air] Remove register-image command and docker_image run support#6657
maggiewang-db wants to merge 1 commit into
mainfrom
air-remove-docker-image

Conversation

@maggiewang-db

@maggiewang-db maggiewang-db commented Sep 11, 2026

Copy link
Copy Markdown
Contributor

Changes

Before the experimental/air release, this PR removes the register-image command and the environment.docker_image run-config path — both replaced by environment.unity_catalog_image.

Removed:

  • The register-image command and its image-registration API client / credential-discovery / docker-config helpers.
  • The environment.docker_image config (type, field, accessors) and the pre-submit image verification / tag re-resolution.
  • The docker_image_url wiring in the submit payload and the validate-config pre-flight.
  • The convert-to-dabs docker_image-not-supported rejection (the field no longer exists).
    • Note: We haven't added convert-to-dabs support for unity_catalog_image yet because the Go SDK in Databricks CLI is stale and missing the unity_catalog_image field. I'll add the convert-to-dabs support once the Go SDK is updated.

Behavior change: environment.unity_catalog_image now inherits the exclusivity docker_image had — it cannot be combined with dependencies or version, since a custom image already pins the full runtime.

Tests

  • Unit tests updated: removed docker_image cases, added unity_catalog_image exclusivity cases and a parse test. go test ./experimental/air/cmd/... passes.
  • Acceptance tests updated: deleted the register-image suites and the convert-to-dabs docker_image case; regenerated help / config-help / convert-to-dabs goldens. go test ./acceptance -run TestAccept/experimental/air passes.
  • go build and go vet ./experimental/air/... are clean.

This pull request and its description were written by Isaac.

Ahead of the experimental/air release, drop the register-image command and the
environment.docker_image run-config path (superseded by
environment.unity_catalog_image). unity_catalog_image now inherits the
'custom image is exclusive with dependencies/version' guard docker_image had.

Removes the register-image command and its image-registration client/credential
helpers, the docker_image config type + accessors, the pre-submit image
verification, the docker_image_url wiring in submit/validate-config, and the
convert-to-dabs docker_image rejection. Updates unit + acceptance tests.

Co-authored-by: Isaac <no-reply@databricks.com>
@github-actions

Copy link
Copy Markdown
Contributor

Approval status: pending

/acceptance/experimental/air/ - needs approval

16 files changed
Suggested: @vinchenzo-db
Also eligible: @ben-hansen-db, @caroline-db, @apeforest, @bfontain, @panchalhp-db, @pardis-beikzadeh-db

/experimental/air/ - needs approval

22 files changed
Suggested: @vinchenzo-db
Also eligible: @ben-hansen-db, @caroline-db, @apeforest, @bfontain, @panchalhp-db, @pardis-beikzadeh-db

Any maintainer (@andrewnester, @anton-107, @denik, @pietern, @shreyas-goenka, @simonfaltum, @renaudhartert-db, @janniklasrose, @lennartkats-db, @rugpanov, @rclarey) can approve all areas.
See OWNERS for ownership rules.

@eng-dev-ecosystem-bot

Copy link
Copy Markdown
Collaborator

Integration test report

Commit: 34d58a5

Run: 34654449797

Env 💚​RECOVERED ✅​pass 🙈​skip Time
💚​ aws linux 1 275 18 4:34
💚​ aws windows 1 277 16 3:46
💚​ azure linux 1 274 18 4:15
💚​ azure windows 1 276 16 3:38
💚​ gcp linux 1 275 18 4:39
💚​ gcp windows 1 277 16 3:38
Test Name aws linux aws windows azure linux azure windows gcp linux gcp windows
💚​ TestAccept 💚​R 💚​R 💚​R 💚​R 💚​R 💚​R
Top 3 slowest tests (at least 2 minutes):
duration env testname
3:43 aws windows TestAccept
3:36 azure windows TestAccept
3:36 gcp windows TestAccept

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants