fix: unit tests

Author: Rohan Gujarathi

src/sagemaker/lambda_helper.py

@@ -31,7 +31,7 @@ def __init__(
         s3_bucket: str = None,
         script: str = None,
         handler: str = None,
-        session: Session = Session(),
+        session: Session = None,
         timeout: int = 120,
         memory_size: int = 128,
         runtime: str = "python3.8",
@@ -73,7 +73,7 @@ def __init__(
         self.script = script
         self.handler = handler
         self.execution_role_arn = execution_role_arn
-        self.session = session
+        self.session = session if session is not None else Session()
         self.timeout = timeout
         self.memory_size = memory_size
         self.runtime = runtime

tests/unit/test_lambda_helper.py

@@ -53,7 +53,7 @@ def s3_client(sagemaker_session):
 
 
 def test_lambda_object_with_arn_happycase():
-    lambda_obj = lambda_helper.Lambda(function_arn=LAMBDA_ARN)
+    lambda_obj = lambda_helper.Lambda(function_arn=LAMBDA_ARN, session=sagemaker_session)
     assert lambda_obj.function_arn == LAMBDA_ARN