From 8c9c240dbb1889a978616c5351009f5edb6493b9 Mon Sep 17 00:00:00 2001
From: Haotian An <haotiaa@amazon.com>
Date: Thu, 25 Apr 2024 02:03:44 +0000
Subject: [PATCH] updates inference component config name

---
 src/sagemaker/jumpstart/session_utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sagemaker/jumpstart/session_utils.py b/src/sagemaker/jumpstart/session_utils.py
index 87c510edaa..0fa7722f91 100644
--- a/src/sagemaker/jumpstart/session_utils.py
+++ b/src/sagemaker/jumpstart/session_utils.py
@@ -55,8 +55,8 @@ def get_model_info_from_endpoint(
             (
                 model_id,
                 model_version,
-                inference_component_name,
                 config_name,
+                inference_component_name,
             ) = _get_model_info_from_inference_component_endpoint_without_inference_component_name(  # noqa E501  # pylint: disable=c0301
                 endpoint_name, sagemaker_session
             )
@@ -70,7 +70,7 @@ def get_model_info_from_endpoint(
 
 def _get_model_info_from_inference_component_endpoint_without_inference_component_name(
     endpoint_name: str, sagemaker_session: Session
-) -> Tuple[str, str, str]:
+) -> Tuple[str, str, str, str]:
     """Derives the model ID, version, config name and inferred inference component name.
 
     This function assumes the endpoint corresponds to an inference-component-based endpoint.