aws / aws/sagemaker-python-sdk

`retrieve` to get lightgbm container returns pytorch inference container

オープン
#4,373 コメント 1 件 リアクション 0 件 担当者 1 名 @nargokul が担当を希望しています GitHub で見る
component: image uri component: pysdk-team type: bug
主要言語
Python
スター
2.3k
フォーク
1.3k
平均マージ
1日 22時間
マージ済み PR(30日)
35

説明

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

コントリビューションガイド

コントリビューションガイドを開く

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。