File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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"
Original file line number Diff line number Diff 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"\n These 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
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 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"
Original file line number Diff line number Diff 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
127127fi
You can’t perform that action at this time.
0 commit comments