aws / aws/sagemaker-python-sdk
Align Remote function classes to ModelTrainer: Include training_plan_arn to remote function classes @remote decorator and RemoteExecutor
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 1d 22h
- Merged PRs (30d)
- 35
Description
**Describe the feature you'd like**
A clear and concise description of the functionality you want.
ModelTrainer and Estimator classes currently support training_plan_arn, in order to run training workload by leveraging the instances included in the plan. This functionality is not currently available with remote function classes, which is adopted by customer to start prototyping workloads on Amazon SageMaker.
**How would this feature be used? Please describe.**
A clear and concise description of the use case for this feature. Please provide an example, if possible.
```
@remote(
...
training_plan_arn="arn.1234....."
)
def train(args):
pass
```
**Describe alternatives you've considered**
A clear and concise description of any alternative solutions or features you've considered.
There are no alternative solutions
**Additional context**
Add any other context or screenshots about the feature request here.
Contributor guide
Research direction
Start by comparing how ModelTrainer and Estimator accept and use training_plan_arn, then trace the @remote decorator and RemoteExecutor paths for equivalent configuration. Define the expected flow for the ARN from decorator arguments into remote execution, and add coverage showing that a remote function can launch with the specified training plan.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- aws, python
- Domain
- cloud, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 45/100