support latest container version in image_uris and DJLModel for lmi containers

siddvenk · siddvenk · commit 1a77e22686ce · 2024-08-15T11:39:32.000-07:00
diff --git a/src/sagemaker/djl_inference/model.py b/src/sagemaker/djl_inference/model.py
@@ -43,7 +43,7 @@ def __init__(
         self,
         model_id: Optional[str] = None,
         engine: Optional[str] = None,
-        djl_version: str = "0.28.0",
+        djl_version: str = "latest",
         djl_framework: Optional[str] = None,
         task: Optional[str] = None,
         dtype: Optional[str] = None,
diff --git a/src/sagemaker/image_uri_config/djl-lmi.json b/src/sagemaker/image_uri_config/djl-lmi.json
@@ -2,6 +2,9 @@
     "scope": [
         "inference"
     ],
+    "version_aliases": {
+        "latest": "0.29.0"
+    },
     "versions": {
         "0.29.0": {
             "registries": {
diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json
@@ -2,6 +2,9 @@
     "scope": [
         "inference"
     ],
+    "version_aliases": {
+        "latest": "0.29.0"
+    },
     "versions": {
         "0.29.0": {
             "registries": {
diff --git a/src/sagemaker/image_uri_config/djl-tensorrtllm.json b/src/sagemaker/image_uri_config/djl-tensorrtllm.json
@@ -2,6 +2,9 @@
     "scope": [
         "inference"
     ],
+    "version_aliases": {
+        "latest": "0.29.0"
+    },
     "versions": {
         "0.29.0": {
             "registries": {
