Hello Gents, I am currently working on Flyte to AWS Sagemaker Integration and facing problems with the idempotence_token in the create_sagemaker_deployment method in the flytekitplugin-awssagemaker_inference plugin version 1.16.1. I am currently testing model, ednpoint config and endpoint deployment using the Flyte Sagemaker plugin and passing the idempotence_token=False in the create_sagemaker_deployment method. But the endpoint_config deployment task still keeps expecting the idempotence_token field in it's input (which is the model_creation task's output). Copilot keeps saying this is known bug and I need to set it to True in order resolve it. But when I set it to True, the model_creation task itself fails in Flyte and gives me an error like so - failed to do boto task with error: Could not find the key model_name}-{idempotence_token in {'model_path': 's3://s3-bucket/models/xgboost-model.tar.gz', 'execution_role_arn': 'arnawsiam::account-id:role/app-flyte-sagemaker-executor-role', 'model_name': 'xgboost-diabetes-endpoint-model'}.. Having a tough time figuring this one out. I have tried multiple approaches but all in vain. Anyone who knows what this is all about?

hey Amit great to meet you

hi @gray-ocean-43286, sorry about the bug you’re running into. when

idempotence_token

is set to

False

, i missed a conditional check, so it was still being added as an input by default. i’ve got a fix for that. for the issue you’re seeing when setting it to

True

, could you share your configs, specifically your

create_sagemaker_deployment

instantiation? also please share the full error stacktrace so i can take a closer look.