aws / aws/aws-step-functions-data-science-sdk-python

Unable set ModelClientConfig in TransformerStep

Open
#171 3 comments 0 reactions 0 assignees View on GitHub
enhancement
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

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.