From 79e061e0e8644a64b6648e79e9ed31f0c1684d41 Mon Sep 17 00:00:00 2001 From: Rohit Kumar Srivastava Date: Wed, 17 Jun 2026 02:38:06 +0000 Subject: [PATCH] updated the SDK to use latest LMI image for sdk v3.x --- sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json | 2 +- sagemaker-core/tests/unit/image_uris/test_djl.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json b/sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json index f76d070464..f0e78267d6 100644 --- a/sagemaker-core/src/sagemaker/core/image_uri_config/djl-lmi.json +++ b/sagemaker-core/src/sagemaker/core/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-lmi26.0.0-cu130" }, "0.35.0": { "registries": { diff --git a/sagemaker-core/tests/unit/image_uris/test_djl.py b/sagemaker-core/tests/unit/image_uris/test_djl.py index 144a68c845..a6416c4d6d 100644 --- a/sagemaker-core/tests/unit/image_uris/test_djl.py +++ b/sagemaker-core/tests/unit/image_uris/test_djl.py @@ -86,6 +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-lmi26.0.0-cu130": {"ap-east-2"}, "0.36.0-lmi22.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"},