From 6c2db19cbd60bf54107f830b4825d0decbc5bbb9 Mon Sep 17 00:00:00 2001 From: smouaa Date: Thu, 2 Apr 2026 17:58:54 +0000 Subject: [PATCH] feat: update SDK to use latest LMI v23 image for sdk v2.x Update tag_prefix from 0.36.0-lmi22.0.0-cu129 to 0.36.0-lmi23.0.0-cu129 for the djl-lmi framework in the image URI config. --- src/sagemaker/image_uri_config/djl-lmi.json | 2 +- tests/unit/sagemaker/image_uris/test_djl.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/image_uri_config/djl-lmi.json b/src/sagemaker/image_uri_config/djl-lmi.json index fcf9b6c9d5..c2e43f41ba 100644 --- a/src/sagemaker/image_uri_config/djl-lmi.json +++ b/src/sagemaker/image_uri_config/djl-lmi.json @@ -46,7 +46,7 @@ "mx-central-1": "637423239942" }, "repository": "djl-inference", - "tag_prefix": "0.36.0-lmi22.0.0-cu129" + "tag_prefix": "0.36.0-lmi23.0.0-cu129" }, "0.35.0": { "registries": { diff --git a/tests/unit/sagemaker/image_uris/test_djl.py b/tests/unit/sagemaker/image_uris/test_djl.py index 95a21639e1..9061e262b9 100644 --- a/tests/unit/sagemaker/image_uris/test_djl.py +++ b/tests/unit/sagemaker/image_uris/test_djl.py @@ -86,7 +86,7 @@ def _test_djl_uris(account, region, version, tag, djl_framework): # Known missing framework:version:region combinations that don't exist in ECR KNOWN_MISSING_COMBINATIONS = { "djl-lmi": { - "0.36.0-lmi22.0.0-cu129": {"ap-east-2"}, + "0.36.0-lmi23.0.0-cu129": {"ap-east-2"}, "0.35.0-lmi17.0.0-cu128": {"ap-east-2"}, "0.34.0-lmi16.0.0-cu128": {"ap-east-2"}, "0.33.0-lmi15.0.0-cu128": {"ap-east-2"},