- Notifications
You must be signed in to change notification settings - Fork89
Open
Description
Not sure if this qualifies as a bug or a feature request...
When creating a transform job one can pass a ModelClientConfig containing the invocation timeout and number of retries. Seehttps://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTransformJob.html#sagemaker-CreateTransformJob-request-ModelClientConfig
This is implemented in the sagemaker python sdk, you can setmodel_client_config when callingTransformer.transform.
There is currently no option to set this in theTransformerStep
It should be relatively easy to implement by addingmodel_client_config as a param to theTransformerStep and settingparameters['ModelClientConfig'] to the value passed there.
This is 🐛 Bug Report