Hello i am trying to work with flyte s `sagemaker custom tra

Question

Hello i am trying to work with flyte s `sagemaker custom training` plugin and when the SageMaker TrainingJob is launched it looks at a `flytekit sagemaker runner py` script as an `ENTRYPOINT` Where can i find this script

Nada Saiyed · Answer

I see that it used to sit in `flytekit` repo under `scripts ` based on this but was removed when plugins were moved out of `flytekit`

Yee · Answer

< Samhita Alla> can you help with this when you get a chance

Ketan (kumare3) · Answer

ohh this is bad i am trying to find when was this file deleted <https github com flyteorg flytekit blob b8a3c7d34e6b19722a6a968dc05506ad1eb26912 flytekit scripts flytekit sagemaker runner py>

Ketan (kumare3) · Answer

< Nada Saiyed> i am adding it back and then will test it

Ketan (kumare3) · Answer

give me some time so sorry This got deleted in some shuffle and sadly we did not catch it in the test

Nada Saiyed · Answer

no worries thanks is the idea that this file will already be present in the ` opt venv bin` dir when trainingJob is launched i e will be in the training image or can i copy it to any directory and give that as an `ENTRYPOINT`

Ketan (kumare3) · Answer

it will be in it after you install the plugin

Ketan (kumare3) · Answer

let me send you a PR if you are around

Ketan (kumare3) · Answer

you can probably help me test it too smile

Nada Saiyed · Answer

sure lmk what i need to do

Ketan (kumare3) · Answer

hey < Nada Saiyed> sorry for the delay here you go

Ketan (kumare3) · Answer

I am testing it now

Ketan (kumare3) · Answer

Also cc < Yee>

Ketan (kumare3) · Answer

I am trying to test it but my role is not correct Just to record you need to set the role in sagemaker config in flytepropeller <https github com flyteorg flyteplugins blob 5a00b19d88b93f9636410a41f81a73356a711482 go tasks plugins k8s sagemaker config config go L37>

Nada Saiyed · Answer

Right i have added the `flytekit sagemaker runner py` in a hacky way in my training image for now and have also updated the config so i am unblocked for now but do let me know when this PR gets merged