Skip to content

Commit 36c7607

Browse files
committed
add public gallery info
Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>
1 parent 77b471d commit 36c7607

File tree

3 files changed

+16
-5
lines changed

3 files changed

+16
-5
lines changed

docs/reference/available_images.template.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
11
# Available Images
22

3-
This page is for referencing our supported deep learning containers. Refer to the tables below for all images that are available in ECR repositories. Additionally, we also publish {{ public_ecr_image_list }} to [ECR Public Gallery](https://gallery.ecr.aws/deep-learning-containers).
3+
This page is for referencing our supported deep learning containers.
4+
5+
Refer to the tables below for all images that are available in AWS ECR repositories.
6+
7+
Additionally, we also publish the following images to [ECR Public Gallery](https://gallery.ecr.aws/deep-learning-containers).
8+
9+
{{ public_ecr_image_list }}
410

511
## Region Availability
612

docs/src/data/images.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,16 +13,16 @@
1313

1414
public_ecr_repositories:
1515
- base
16+
- sglang
1617
- vllm
1718
- vllm-arm64
18-
- sglang
1919
- pytorch-training
2020
- pytorch-inference
21+
- pytorch-training-arm64
22+
- pytorch-inference-arm64
2123
- tensorflow-training
2224
- tensorflow-inference
23-
- pytorch-inference-arm64
2425
- tensorflow-inference-arm64
25-
- pytorch-training-arm64
2626

2727
# -----------------------------------------------------------------------------
2828
# Framework Support Policy

docs/src/macros.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,12 @@ def define_env(env):
3636
"""Define variables for mkdocs-macros-plugin."""
3737
data = load_yaml(DATA_FILE)
3838
public_repos = data.get("public_ecr_repositories", [])
39-
env.variables["public_ecr_image_list"] = ", ".join(public_repos)
39+
40+
# Assign environment at mkdocs build time
41+
env.variables["public_ecr_image_list"] = "\n".join(
42+
f"- [{repo}](https://gallery.ecr.aws/deep-learning-containers/{repo})"
43+
for repo in public_repos
44+
)
4045
env.variables["images"] = {
4146
"latest_pytorch_training_ec2": get_latest_image("pytorch-training", "-ec2"),
4247
"latest_vllm_sagemaker": get_latest_image("vllm", "-sagemaker"),

0 commit comments

Comments
 (0)