aws / aws/sagemaker-python-sdk

`retrieve` to get lightgbm container returns pytorch inference container

Đang mở
#4,373 1 bình luận 0 reaction 1 người được giao Được @nargokul nhận Xem trên GitHub
component: image uri component: pysdk-team type: bug
Ngôn ngữ chính
Python
Star
2.3k
Fork
1.3k
Merge trung bình
1 ngày 22 giờ
Pull request đã merge (30 ngày)
35

Mô tả

**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.

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Đánh giá

Issue này chưa được đánh giá.

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.