aws / aws/sagemaker-python-sdk
Can I create lambda step from ecr image by using Lambda class in lambda_help.py?
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 2d 2h
- Merged PRs (30d)
- 32
Description
**Describe the feature you'd like**
Hey! As I read from the code in lambda_helper.py, I only see Lambda class wrapped creating lambda function from a script. Is there any reason it does not support creating function from ecr images?
**How would this feature be used? Please describe.**
should be a parameter that I can refer to the image uri:
```
Lambda(
function_name="function_name",
execution_role_arn=params["pipeline_steps_arn"],
image="image_uri"
handler=step_config["handler"],
session=session,
timeout=step_config["timeout"],
memory_size=step_config["memory_size"],
runtime=step_config["runtime"],
)
```
Currently, I did in this way (from boto3 lambda client call):
```
response = lambda_client.create_function(
FunctionName=function_name,
Role=params["pipeline_steps_arn"],
Code={"ImageUri": step_config["image_uri"]},
PackageType="Image",
Timeout=step_config["timeout"],
MemorySize=step_config["memory_size"],
Architectures=["x86_64"],
)
return Lambda(function_arn=response["FunctionArn"], session=session)
```
Contributor guide
Assessment
This issue has not been assessed yet.