Skip to content

Commit 0a4e745

Browse files
authored
Merge branch 'main' into omni
2 parents b1d1eac + 6c78b54 commit 0a4e745

5 files changed

Lines changed: 29 additions & 1 deletion

File tree

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
framework: XGBoost
2+
version: "3.0-5"
3+
accelerator: cpu
4+
platform: sagemaker
5+
example_ecr_account: "<account-id>"
6+
7+
tags:
8+
- "3.0-5"
9+
10+
ga: "2025-11-17"
11+
eop: "2026-11-17"

docs/src/generate.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,8 @@ def _collapse_minor_versions(
246246
for mm, indices in collapsible_groups.items():
247247
if mm in uncollapsible:
248248
continue
249+
if len(indices) < 2:
250+
continue
249251
group_imgs = [entries[idx][0] for idx in indices]
250252
ref_img = group_imgs[0]
251253
if dates_agree(group_imgs):
@@ -378,6 +380,8 @@ def generate_available_images(dry_run: bool = False) -> str:
378380
section += (
379381
f"\nThese images are also available in ECR Public Gallery: [{repository}]({url})\n"
380382
)
383+
if table_config.get("note"):
384+
section += f"\n{table_config['note']}\n"
381385
section += f"\n{render_table(headers, rows)}"
382386
tables_content.append(section)
383387

docs/src/global.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ display_names:
6262

6363
# Repositories
6464
base: "Base"
65+
sagemaker-xgboost: "XGBoost"
6566
sglang: "SGLang"
6667
vllm: "vLLM"
6768
vllm-arm64: "vLLM ARM64"
@@ -185,3 +186,4 @@ table_order:
185186
- sagemaker-tritonserver
186187
- stabilityai-pytorch-inference
187188
- pytorch-trcomp-training
189+
- sagemaker-xgboost
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
# Table Configuration - XGBoost
2+
note: "ECR account IDs for XGBoost vary by region. Refer to [SageMaker AI ECR Paths](https://docs.aws.amazon.com/sagemaker/latest/dg-ecr-paths/sagemaker-algo-docker-registry-paths.html) for your region's account ID."
3+
columns:
4+
- field: framework_version
5+
header: "Framework"
6+
- field: accelerator
7+
header: "Accelerator"
8+
- field: platform
9+
header: "Platform"
10+
- field: example_url
11+
header: "Example URL"

scripts/autocurrency/docs-pr.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ IMAGE_URI="public.ecr.aws/deep-learning-containers/${FRAMEWORK}:${VERSION}-${DEV
121121
# -----------------------------------------------------------------
122122
# Early exit: skip unsupported platforms
123123
# -----------------------------------------------------------------
124-
if [ "$PLATFORM" = "rayserve_ec2" ]; then
124+
if [ "$PLATFORM" = "rayserve_ec2" ] || [ "$FRAMEWORK" = "xgboost" ]; then
125125
echo "${FRAMEWORK}: Platform '${PLATFORM}' is not supported for docs generation. Skipping."
126126
exit 0
127127
fi

0 commit comments

Comments
 (0)