aws / aws/sagemaker-python-sdk

Align Remote function classes to ModelTrainer: Include training_plan_arn to remote function classes @remote decorator and RemoteExecutor

Open
#5,157 0 comments 0 reactions 0 assignees View on GitHub
remote-function type: feature request
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.