aws / aws/sagemaker-python-sdk

Can I create lambda step from ecr image by using Lambda class in lambda_help.py?

Open
#4,819 0 comments 1 reaction 1 assignee Claimed by @mollyheamazon View on GitHub
component: pipelines type: feature request
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.