Alternatively, you can try this one - https://github.com/aws-samples/sagemaker-run-notebook. It uses papermill to run a jupyter notebook in an AWS SageMaker Processing Job. The job is triggered by a lambda function. This creates an alternative in case your job runs past the 15 minutes timeout mark of a lambda function.