aws / aws/aws-step-functions-data-science-sdk-python
Unable set ModelClientConfig in TransformerStep
- Dominant language
- Python
- Stars
- 299
- Forks
- 84
- PR merge metrics
- No merged PRs in 30d
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. See https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTransformJob.html#sagemaker-CreateTransformJob-request-ModelClientConfig
This is implemented in the sagemaker python sdk, you can set `model_client_config` when calling `Transformer.transform`.
There is currently no option to set this in the `TransformerStep`
It should be relatively easy to implement by adding `model_client_config` as a param to the `TransformerStep` and setting `parameters['ModelClientConfig']` to the value passed there.
---
This is :bug: Bug Report
Contributor guide
Assessment
This issue has not been assessed yet.