Add Lambda Layer argument

Author: dipanjank

src/sagemaker/lambda_helper.py

@@ -39,6 +39,7 @@ def __init__(
         vpc_config: dict = None,
         architectures: list = None,
         environment: dict = None,
+        layers: list = None,
     ):
         """Constructs a Lambda instance.
 
@@ -73,6 +74,7 @@ def __init__(
             architectures (list): Which architecture to deploy to. Valid Values are
                 'x86_64' and 'arm64', default is None.
             environment (dict): Environment Variables for the Lambda function. Default is None.
+            layers (list): List of Lambda layers for the Lambda function. Default is None.
         """
         self.function_arn = function_arn
         self.function_name = function_name
@@ -88,6 +90,7 @@ def __init__(
         self.vpc_config = vpc_config
         self.environment = environment
         self.architectures = architectures
+        self.layers = layers
 
         if function_arn is None and function_name is None:
             raise ValueError("Either function_arn or function_name must be provided.")
@@ -140,6 +143,7 @@ def create(self):
                 VpcConfig=self.vpc_config,
                 Environment=self.environment,
                 Architectures=self.architectures,
+                Layers=self.layers,
             )
             return response
         except ClientError as e:

tests/unit/test_lambda_helper.py

@@ -190,6 +190,7 @@ def test_create_lambda_happycase1(sagemaker_session):
         Architectures=None,
         VpcConfig=None,
         Environment=None,
+        Layers=None,
     )
 
 
@@ -218,6 +219,7 @@ def test_create_lambda_happycase2(sagemaker_session):
         Architectures=None,
         VpcConfig=None,
         Environment=None,
+        Layers=None,
     )
 
 
@@ -235,6 +237,7 @@ def test_create_lambda_happycase3(sagemaker_session):
             "SubnetIds": ["test-subnet-1"],
             "SecurityGroupIds": ["sec-group-1"],
         },
+        layers=["my-test-layer-1", "my-test-layer-2"],
     )
 
     lambda_obj.create()
@@ -251,6 +254,7 @@ def test_create_lambda_happycase3(sagemaker_session):
         Architectures=["x86_64"],
         VpcConfig={"SubnetIds": ["test-subnet-1"], "SecurityGroupIds": ["sec-group-1"]},
         Environment={"Name": "my-test-lambda"},
+        Layers=["my-test-layer-1", "my-test-layer-2"],
     )
 
 
@@ -424,6 +428,7 @@ def test_upsert_lambda_happycase1(sagemaker_session):
         Architectures=None,
         VpcConfig=None,
         Environment=None,
+        Layers=None,
     )