Skip to content

retrieve to get lightgbm container returns pytorch inference container #4373

Open
@ravinkohli

Description

@ravinkohli

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.

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions