Skip to content

air get: overlap the text view's run-only lookups with get-output - #7626

Open
stevenchen-db wants to merge 2 commits into
mainfrom
air/get-overlap-lookups
Open

stevenchen-db wants to merge 2 commits into
mainfrom
air/get-overlap-lookups

Conversation

@stevenchen-db

Copy link
Copy Markdown
Contributor

Changes

air get in text mode made five requests one after another: runs/get, get-output, a second runs/get for the serverless environment version (the typed SDK Run has no environments), the MLflow run name, and the training-config export. The environment lookup and the config export depend only on the run, so they now start right after the first runs/get and overlap get-output; the MLflow name lookup starts before waiting on them. The list picker's detail pane uses the same renderer and gets the same overlap. JSON output is unchanged (it makes none of these lookups).

Why

On df1, air get <run> (text) went from ~536-581ms to ~380-404ms median (10 runs per build, interleaved).

Tests

  • New TestGetTextOverlapsRunOnlyLookupsWithGetOutput: the server answers get-output only after the config export and the second runs/get arrive; it fails on main.
  • go test ./cmd/air (touched tests also with -race), TestAccept/air, lint pass.
  • df1 vs main: air get text and JSON output byte-identical on three runs; JSON makes the same two requests; the air list detail pane renders the same with the same requests.

This pull request and its description were written by Isaac.

`air get` in text mode made five requests one after another: runs/get,
get-output, a second runs/get for the serverless environment version,
the MLflow run name, and the training-config export. The environment
lookup and the config export depend only on the run, so start them right
after the first runs/get and let them overlap get-output; start the MLflow
name lookup before waiting on them. The list picker's detail pane shares
the renderer and gets the same overlap.

Co-authored-by: Isaac <no-reply@databricks.com>
@stevenchen-db
stevenchen-db requested review from a team as code owners November 14, 2026 00:33
Co-authored-by: Isaac <no-reply@databricks.com>
@github-actions github-actions Bot added the AIR Databricks AI Runtime CLI label Nov 14, 2026
@eng-dev-ecosystem-bot

Copy link
Copy Markdown
Collaborator

✅ Integration test report

Commit: dcad2e5

Run: 43535239798

Env 💚​RECOVERED ✅​pass 🙈​skip Time
💚​ aws linux-2core-8gb 1 270 15 5:08
💚​ aws-windows-latest-4core-16gb 1 272 13 4:56
💚​ azure linux-2core-8gb 1 269 15 5:22
💚​ azure-windows-latest-4core-16gb 1 271 13 4:17
💚​ gcp linux-2core-8gb 1 270 15 6:36
💚​ gcp-windows-latest-4core-16gb 1 272 13 4:55
Test Name aws linux-2core-8gb aws-windows-latest-4core-16gb azure linux-2core-8gb azure-windows-latest-4core-16gb gcp linux-2core-8gb gcp-windows-latest-4core-16gb
💚​ TestAccept 💚​R 💚​R 💚​R 💚​R 💚​R 💚​R
Top 3 slowest tests (at least 2 minutes):
duration env testname
3:27 gcp-windows-latest-4core-16gb TestAccept
3:22 azure-windows-latest-4core-16gb TestAccept
3:17 aws-windows-latest-4core-16gb TestAccept

This branch has not been deployed

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

Labels

AIR Databricks AI Runtime CLI

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants