Describe the bug
Currently image_uri retrieval always uses default jumpstart. For those who wants to use that function and passes a sagemaker session with profiles setup , it will not be received when running image_uris.retrieve(). the session is currently only used to retrieve the region of the session only
To reproduce
A clear, step-by-step set of instructions to reproduce the bug.
Expected behavior
sagemaker-python-sdk/src/sagemaker/huggingface
if backend == "huggingface":
return image_uris.retrieve(
"huggingface-llm",
region=region,
version=version,
image_scope="inference",
sagemaker_session=session
)
.