aws-step-functions-data-science-sdk-python icon indicating copy to clipboard operation
aws-step-functions-data-science-sdk-python copied to clipboard

Support ModelPackage in the ModelStep

Open shunjd opened this issue 5 years ago • 0 comments
trafficstars

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

shunjd avatar Jul 21 '20 16:07 shunjd