aws / aws/sagemaker-python-sdk
`retrieve` to get lightgbm container returns pytorch inference container
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 1d 22h
- Merged PRs (30d)
- 35
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.
```python
# 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.
Contributor guide
Assessment
This issue has not been assessed yet.