Hello.. i am trying to work with flyte’s `sagemake...
# ask-the-community
n
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
@Samhita Alla can you help with this when you get a chance?
k
@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
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
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
sure. lmk what i need to do
k
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
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.
135 Views