Skip to content

feat: add Azure Linux image family support with docs and e2e tests#294

Merged
Fei-Guo merged 2 commits into
mainfrom
mitchzhu/azure-linux-support-rebased
Feb 13, 2026
Merged

feat: add Azure Linux image family support with docs and e2e tests#294
Fei-Guo merged 2 commits into
mainfrom
mitchzhu/azure-linux-support-rebased

Conversation

@miz060
Copy link
Copy Markdown
Member

@miz060 miz060 commented Feb 11, 2026

Rebased from: #250
This pull request introduces support for provisioning Azure Linux nodes in the GPU Provisioner and updates related functionality across documentation, implementation, and testing. Key changes include adding configuration options for Azure Linux in KAITO workloads, modifying the logic for determining the OS SKU, and expanding test coverage to ensure compatibility.

Documentation Updates:

  • Added Azure Linux examples under examples/:
    • azure-linux-nodeclaim.yaml
    • azure-linux-annotation-nodeclaim.yaml
    • azure-linux-examples.md

Implementation Enhancements:

  • pkg/providers/instance/instance.go: determine OSSKU from NodeClaim labels/annotations (labels
    take precedence; default to Ubuntu).

Testing Improvements:

  • Added unit test TestNewAgentPoolObjectWithImageFamily in
    pkg/providers/instance/instance_test.go.
  • Updated e2e suite in test/e2e/suites/suite_test.go:
    • Added Azure Linux e2e coverage.

Validation:

  • Manual validation in addition to e2e: created an AKS cluster with H100/A100, built and pushed an AzureLinux‑enabled gpu‑provisioner image to ACR, configured managed identity with contributor on the cluster RG, deployed the gpu‑provisioner via Helm, then created a Workspace with AzureLinux + GPU + the phi‑3‑mini model and verified the full flow worked).

@miz060 miz060 requested a review from rambohe-ch February 11, 2026 18:26
@miz060 miz060 changed the title feat: add Azure Linux image family support with docs and e2e tests feat: add Azure Linux image family support with docs and e2e tests Feb 11, 2026
@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Feb 11, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 71.39%. Comparing base (d464f29) to head (2b3d534).

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #294      +/-   ##
==========================================
+ Coverage   70.07%   71.39%   +1.31%     
==========================================
  Files           4        4              
  Lines         391      409      +18     
==========================================
+ Hits          274      292      +18     
  Misses         98       98              
  Partials       19       19              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

- add Azure Linux image family support
- add Azure Linux NodeClaim examples and doc updates
- add e2e suite coverage

Signed-off-by: miz060 <mitchzhu@microsoft.com>
@miz060 miz060 force-pushed the mitchzhu/azure-linux-support-rebased branch from 7979e6b to 74241a9 Compare February 11, 2026 18:57
Comment thread test/e2e/suites/suite_test.go Outdated
Comment thread test/e2e/suites/suite_test.go Outdated
Comment thread test/e2e/suites/suite_test.go
@miz060
Copy link
Copy Markdown
Member Author

miz060 commented Feb 13, 2026

E2E tests failed because of azure security policy.
I have run full E2E locally and confirmed all tests passed.

E2E-passed

@rambohe-ch rambohe-ch enabled auto-merge (squash) February 13, 2026 10:09
@Fei-Guo Fei-Guo disabled auto-merge February 13, 2026 16:28
@Fei-Guo Fei-Guo merged commit 01b8b54 into main Feb 13, 2026
5 of 6 checks passed
@Fei-Guo Fei-Guo deleted the mitchzhu/azure-linux-support-rebased branch February 13, 2026 16:28
@rambohe-ch
Copy link
Copy Markdown
Collaborator

Fixes #158 #255

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.

4 participants