Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Flyte

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?

I see that it used to sit in `flytekit` repo under `scripts/` based on this <https://github.com/flyteorg/flytekit/pull/156/files#diff-5278c0a55184058d80b91085acfa26e04892a04c0fb1e7871c0f39d4620dd192|PR> but was removed when plugins were moved out of `flytekit`  (<https://github.com/flyteorg/flytekit/commit/ba29604fdf8751610a8f52c1cf99521e2631eeca|commit>)

<@U01J90KBSU9> can you help with this when you get a chance?

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>

<@U03KZJ7AXKR> i am adding it back and then will test it

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

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` ?

it will be in it after you install the plugin

you can probably help me test it too :smile:

hey <@U03KZJ7AXKR> sorry for the delay - here you go - <https://github.com/flyteorg/flytekit/pull/1112>

Also <https://github.com/flyteorg/flytesnacks/pull/842>
cc <@UNR3C6Y4T>

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>

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.