https://flyte.org logo
#ask-the-community
Title
# ask-the-community
n

Nada Saiyed

07/28/2022, 7:09 PM
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 PR but was removed when plugins were moved out of
flytekit
(commit)
y

Yee

07/28/2022, 8:08 PM
@Samhita Alla can you help with this when you get a chance?
k

Ketan (kumare3)

07/28/2022, 9:16 PM
@Nada Saiyed 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
n

Nada Saiyed

07/28/2022, 9:25 PM
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
?
k

Ketan (kumare3)

07/28/2022, 9:29 PM
it will be in it after you install the plugin
let me send you a PR if you are around
you can probably help me test it too 😄
n

Nada Saiyed

07/28/2022, 9:29 PM
sure. lmk what i need to do
k

Ketan (kumare3)

07/28/2022, 10:47 PM
hey @Nada Saiyed sorry for the delay - here you go - https://github.com/flyteorg/flytekit/pull/1112
I am testing it now
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
n

Nada Saiyed

08/02/2022, 5:39 PM
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.
25 Views