Open
Description
Describe the bug
I am trying to retrieve a container for the lightgbm-classification-model
with version 2.1.0
. However, when calling retrieve
I get a docker container for the pytorch inference script.
To reproduce
The provided code need to be complete and runnable, if additional data is needed, please include them in the issue.
# Create a Sagemaker model
from sagemaker.image_uris import retrieve
#### Get the built-in xgboost container image in Sagemaker to host our model
train_model_id, train_model_version, train_scope = "lightgbm-classification-model", "2.1.0", "inference"
training_instance_type = "ml.m5.xlarge"
# Retrieve the docker image
container = retrieve(
region=None,
framework=None,
model_id=train_model_id,
model_version=train_model_version,
image_scope=train_scope,
instance_type=training_instance_type
)
print(container)
Output is '763104351884.dkr.ecr.eu-central-1.amazonaws.com/pytorch-inference:2.0.1-cpu-py310'
Expected behavior
I expected a docker image for lightgbm model to be returned.
Screenshots or logs
If applicable, add screenshots or logs to help explain your problem.
System information
A description of your system. Please provide:
- SageMaker Python SDK version: 2.203.1
- Framework name (eg. PyTorch) or algorithm (eg. KMeans):
LightGBM
- Framework version: 2.1.0 (sagemaker version, lib version 4.2.0)
- Python version: 3.9
- CPU or GPU: CPU
- Custom Docker image (Y/N): N
Additional context
Add any other context about the problem here.