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

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.