Hey team , I am trying to implement a lambda step within my sagemaker pipeline as follows:

# Custom Lambda Step
current_time = time.strftime("%m-%d-%H-%M-%S", time.localtime())
model_name = "demo-lambda-model" + current_time
endpoint_config_name = "demo-lambda-deploy-endpoint-config-" + current_time
endpoint_name = "demo-lambda-deploy-endpoint-" + current_time

function_name = "sagemaker-lambda-step-endpoint-deploy"
model_package_group_name = "MLOpsModelPackageGroup"

# Lambda helper class can be used to create the Lambda function
func = Lambda(
    function_name=function_name,
    execution_role_arn=role,
    script="pipelines/abalone/inferencing_lambda_helper.py",
    handler="inferencing_lambda_helper.lambda_handler",
)

output_param_1 = LambdaOutput(output_name="statusCode", output_type=LambdaOutputTypeEnum.String)
output_param_2 = LambdaOutput(output_name="body", output_type=LambdaOutputTypeEnum.String)
output_param_3 = LambdaOutput(output_name="other_key", output_type=LambdaOutputTypeEnum.String)

step_lambda = LambdaStep(
    name="LambdaStep",
    lambda_func=func,
    inputs={
        "MODEL_GROUP_NAME": model_package_group_name,
        "ROLE": role,
        "model_name": model_name,  #step_create_model.properties.ModelName,
        "endpoint_config_name": endpoint_config_name,
        "endpoint_name": endpoint_name,
    },
    outputs=[output_param_1, output_param_2, output_param_3],
)

I did import these:

from sagemaker.lambda_helper import Lambda

from sagemaker.workflow.lambda_step import ( LambdaStep, LambdaOutput, LambdaOutputTypeEnum, )

using Pipelinesession():

sagemaker_session = PipelineSession()

Error:

on the pipeline step log I am getting this error: Unable to import module 'lambda_helper': No module named 'lambda_helper'

Question:

any idea to what I am doing wrong? Please point me in the right direction

Thank you!