Skip to content

Xgboost migration fixes#5892

Merged
Jyothirmaikottu merged 10 commits into
mainfrom
xgboost-migration-fixes
Apr 7, 2026
Merged

Xgboost migration fixes#5892
Jyothirmaikottu merged 10 commits into
mainfrom
xgboost-migration-fixes

Conversation

@Jyothirmaikottu
Copy link
Copy Markdown
Contributor

Purpose

Test Plan

Test Result


Toggle if you are merging into master Branch

By default, docker image builds and tests are disabled. Two ways to run builds and tests:

  1. Using dlc_developer_config.toml
  2. Using this PR description (currently only supported for PyTorch, TensorFlow, vllm, and base images)
How to use the helper utility for updating dlc_developer_config.toml

Assuming your remote is called origin (you can find out more with git remote -v)...

  • Run default builds and tests for a particular buildspec - also commits and pushes changes to remote; Example:

python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -cp origin

  • Enable specific tests for a buildspec or set of buildspecs - also commits and pushes changes to remote; Example:

python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -t sanity_tests -cp origin

  • Restore TOML file when ready to merge

python src/prepare_dlc_dev_environment.py -rcp origin

NOTE: If you are creating a PR for a new framework version, please ensure success of the local, standard, rc, and efa sagemaker tests by updating the dlc_developer_config.toml file:

  • sagemaker_remote_tests = true
  • sagemaker_efa_tests = true
  • sagemaker_rc_tests = true
  • sagemaker_local_tests = true
How to use PR description Use the code block below to uncomment commands and run the PR CodeBuild jobs. There are two commands available:
  • # /buildspec <buildspec_path>
    • e.g.: # /buildspec pytorch/training/buildspec.yml
    • If this line is commented out, dlc_developer_config.toml will be used.
  • # /tests <test_list>
    • e.g.: # /tests sanity security ec2
    • If this line is commented out, it will run the default set of tests (same as the defaults in dlc_developer_config.toml): sanity, security, ec2, ecs, eks, sagemaker, sagemaker-local.
# /buildspec <buildspec_path>
# /tests <test_list>
Toggle if you are merging into main Branch

PR Checklist

  • [] I ran pre-commit run --all-files locally before creating this PR. (Read DEVELOPMENT.md for details).

---
X-AI-Tool: Kiro-cli
X-AI-Handle-Time-Seconds: 152
X-AI-Prompt: the tags of xgboost are 3.0-5, 3.0-5-1.0.15.0, 3.0-5-cpu-py3, we need to maintain this in the repo config
---
X-AI-Tool: Kiro-cli
X-AI-Handle-Time-Seconds: 79
X-AI-Prompt: remove these changes its not need, we need to modify the ci tag to not have sagemaker init
---
X-AI-Tool: Kiro-cli
X-AI-Handle-Time-Seconds: 19
X-AI-Prompt: remove cuda, os version
---
X-AI-Tool: Kiro-cli
X-AI-Handle-Time-Seconds: 51
X-AI-Prompt: chekc the workflows and make modifications
fix: simplify XGBoost CI tags and fix integ test workflow ordering

Simplify the CI image tags for XGBoost PR and release workflows by
removing the cuda version, OS version, and sagemaker suffix. The
tags now follow the pattern xgboost-{version}-{device}-{python}-{id}
which aligns with the public release tag scheme.

Fix the integration test workflow so that E2E tests depend on
container tests completing first, matching the documented execution
order in the README. Remove stale comments that incorrectly stated
container and benchmark tests were commented out. Correct the README
to reflect that the release workflow is triggered only by
workflow_dispatch, not push.

ai-dev-branch commit IDs:
6ba7a2c
cb1bc7c
d39feb9
c9ffef7
The prompts used are captured in the footers of those commits.
The initial prompt was: the tags of xgboost are 3.0-5, 3.0-5-1.0.15.0, 3.0-5-cpu-py3, we need to maintain this in the repo config

---
X-AI-Handle-Time-Seconds: 301
X-AI-Line-Changes: New:15, Altered:7, Deleted:14
X-Human-Line-Changes: New:0, Altered:0, Deleted:0
X-AI-Line-Changes-Kiro-cli: New:15, Altered:7, Deleted:14
X-AI-Handle-Time-Seconds-Kiro-cli: 301
X-AI-Change-Count: 4
X-Human-Change-Count: 0
X-AI-Change-Count-Kiro-cli: 4
X-CR-Amendment: false
@aws-deep-learning-containers-ci aws-deep-learning-containers-ci Bot added authorized Size:XS Determines the size of the PR labels Apr 7, 2026
Jyothirmaikottu and others added 3 commits April 7, 2026 14:12
---
X-AI-Tool: Kiro-cli
X-AI-Handle-Time-Seconds: 318
X-AI-Prompt: so there are precommit checks which is a job that checks the syntax, when i raised a pr it passes but for my teammates the xgboost files fail in pre commit checks why is this happening
fix: simplify XGBoost CI tags, fix integ test ordering, and resolve typos pre-commit failure

Simplify the CI image tags for XGBoost PR and release workflows by
removing the cuda version, OS version, and sagemaker suffix. The
tags now follow the pattern xgboost-{version}-{device}-{python}-{id}
which aligns with the public release tag scheme.

Fix the integration test workflow so that E2E tests depend on
container tests completing first, matching the documented execution
order in the README. Remove stale comments that incorrectly stated
container and benchmark tests were commented out. Correct the README
to reflect that the release workflow is triggered only by
workflow_dispatch.

Add certifi to the typos allow list in _typos.toml to fix
pre-commit failures on all PRs caused by the typos checker
flagging the Python package name in the XGBoost Dockerfile.

ai-dev-branch commit IDs:
6ba7a2c
cb1bc7c
d39feb9
c9ffef7
39ca9b1
The prompts used are captured in the footers of those commits.
The initial prompt was: the tags of xgboost are 3.0-5, 3.0-5-1.0.15.0, 3.0-5-cpu-py3, we need to maintain this in the repo config

---
X-AI-Handle-Time-Seconds: 318
X-AI-Line-Changes: New:1, Altered:0, Deleted:0
X-Human-Line-Changes: New:0, Altered:0, Deleted:0
X-AI-Line-Changes-Kiro-cli: New:1, Altered:0, Deleted:0
X-AI-Handle-Time-Seconds-Kiro-cli: 318
X-AI-Change-Count: 1
X-Human-Change-Count: 0
X-AI-Change-Count-Kiro-cli: 1
X-CR-Amendment: false
@Jyothirmaikottu Jyothirmaikottu enabled auto-merge (squash) April 7, 2026 21:53
@Jyothirmaikottu Jyothirmaikottu merged commit 68952b8 into main Apr 7, 2026
10 checks passed
@junpuf junpuf deleted the xgboost-migration-fixes branch April 14, 2026 17:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

authorized Size:XS Determines the size of the PR

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants