aws / aws/aws-step-functions-data-science-sdk-python

Support ModelPackage in the ModelStep

Open
#67 0 comments 2 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
299
Forks
84
PR merge metrics
No merged PRs in 30d

Description

`ModelStep` does not handle `ModelPackage` correctly, we need a branching logic to set the correct parameters when the model is an instance of `ModelPackage`.

ModelPackage: https://github.com/aws/sagemaker-python-sdk/blob/1ff8bd623dc9a13cb38f8253d098a5fffee29833/src/sagemaker/model.py#L903

Contributor guide

Open the contributing guide

Research direction

Start by locating the ModelStep implementation and reading the linked ModelPackage definition in the SageMaker Python SDK. Compare the parameters ModelStep currently handles with those required for a ModelPackage; the work is done when ModelPackage inputs receive the correct parameters.

Written by the indexing model from the issue text.

Assessment

Tech stack
aws, python
Domain
cloud, machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.