https://flyte.org
Join Slack
Hello.. i am trying to work with flyte’s `sagemaker_custom_training` plugin and when the SageMaker T...
# flyte-support
p
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)
t
@tall-lock-23197 can you help with this when you get a chance?
f
@plain-carpenter-67621 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
p
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
?
f
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 😄
p
sure. lmk what i need to do
f
hey @plain-carpenter-67621 sorry for the delay - here you go - https://github.com/flyteorg/flytekit/pull/1112
I am testing it now
Also https://github.com/flyteorg/flytesnacks/pull/842 cc @thankful-minister-83577
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
p
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.
158 Views
Open in Slack
PreviousNext
Powered by

Flyte

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

Powered by