aws-step-functions-data-science-sdk-python
aws-step-functions-data-science-sdk-python copied to clipboard
Support ModelPackage in the ModelStep
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