Skip to content

Commit c6cff06

Browse files
committed
added missing framework:version:region combinations to fix failing DJL tests ref aws#5509
1 parent 27b8517 commit c6cff06

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

tests/unit/sagemaker/image_uris/test_djl.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,12 @@ def _test_djl_uris(account, region, version, tag, djl_framework):
8686
# Known missing framework:version:region combinations that don't exist in ECR
8787
KNOWN_MISSING_COMBINATIONS = {
8888
"djl-lmi": {
89+
"0.36.0-lmi18.0.0-cu128": {"ap-east-2"},
90+
"0.35.0-lmi17.0.0-cu128": {"ap-east-2"},
91+
"0.34.0-lmi16.0.0-cu128": {"ap-east-2"},
92+
"0.33.0-lmi15.0.0-cu128": {"ap-east-2"},
93+
"0.32.0-lmi14.0.0-cu126": {"ap-east-2"},
94+
"0.31.0-lmi13.0.0-cu124": {"ap-east-2"},
8995
"0.30.0-lmi12.0.0-cu124": {"ap-east-2"},
9096
"0.29.0-lmi11.0.0-cu124": {"ap-east-2"},
9197
"0.28.0-lmi10.0.0-cu124": {"ap-east-2"},
@@ -205,6 +211,8 @@ def _test_djl_uris(account, region, version, tag, djl_framework):
205211
},
206212
},
207213
"djl-tensorrtllm": {
214+
"0.33.0-tensorrtllm0.21.0-cu128": {"ap-east-2"},
215+
"0.32.0-tensorrtllm0.12.0-cu125": {"ap-east-2"},
208216
"0.30.0-tensorrtllm0.12.0-cu125": {"ap-east-2"},
209217
"0.29.0-tensorrtllm0.11.0-cu124": {"ap-east-2"},
210218
"0.28.0-tensorrtllm0.9.0-cu122": {"ap-east-2"},

0 commit comments

Comments
 (0)