change: Update contents of key pipeline notebooks based on latest SDK…

… update (aws#3499) * change: Update contents of key pipeline notebooks based on latest SDK update * Add updated output notebooks Co-authored-by: Dewen Qi <[email protected]>
hchings · Jul 14, 2022 · b7efbbc · b7efbbc
1 parent d391107
commit b7efbbc
Show file tree

Hide file tree

Showing 8 changed files with 2,510 additions and 2,807 deletions.
diff --git a/...ne_build_train_deploy/sagemaker-pipelines-preprocess-train-evaluate-batch-transform.ipynb b/...ne_build_train_deploy/sagemaker-pipelines-preprocess-train-evaluate-batch-transform.ipynb
diff --git a/..._train_deploy/sagemaker-pipelines-preprocess-train-evaluate-batch-transform_outputs.ipynb b/..._train_deploy/sagemaker-pipelines-preprocess-train-evaluate-batch-transform_outputs.ipynb
diff --git a/sagemaker-pipelines/tabular/lambda-step/iam_helper.py b/sagemaker-pipelines/tabular/lambda-step/iam_helper.py
@@ -1,42 +1,42 @@
 import boto3
 import json
 
-iam = boto3.client('iam')
+iam = boto3.client("iam")
+
 
 def create_lambda_role(role_name):
     try:
         response = iam.create_role(
-            RoleName = role_name,
-            AssumeRolePolicyDocument = json.dumps({
-                "Version": "2012-10-17",
-                "Statement": [
-                    {
-                        "Effect": "Allow",
-                        "Principal": {
-                            "Service": "lambda.amazonaws.com"
-                        },
-                        "Action": "sts:AssumeRole"
-                    }
-                ]
-            }),
-            Description='Role for Lambda to call SageMaker functions'
+            RoleName=role_name,
+            AssumeRolePolicyDocument=json.dumps(
+                {
+                    "Version": "2012-10-17",
+                    "Statement": [
+                        {
+                            "Effect": "Allow",
+                            "Principal": {"Service": "lambda.amazonaws.com"},
+                            "Action": "sts:AssumeRole",
+                        }
+                    ],
+                }
+            ),
+            Description="Role for Lambda to call SageMaker functions",
         )
 
-        role_arn = response['Role']['Arn']
+        role_arn = response["Role"]["Arn"]
 
         response = iam.attach_role_policy(
             RoleName=role_name,
-            PolicyArn='arn:aws:iam::aws:policy/service-role/AWSLambdaBasicExecutionRole'
+            PolicyArn="arn:aws:iam::aws:policy/service-role/AWSLambdaBasicExecutionRole",
         )
 
         response = iam.attach_role_policy(
-            PolicyArn='arn:aws:iam::aws:policy/AmazonSageMakerFullAccess',
-            RoleName=role_name
+            PolicyArn="arn:aws:iam::aws:policy/AmazonSageMakerFullAccess", RoleName=role_name
         )
 
         return role_arn
 
     except iam.exceptions.EntityAlreadyExistsException:
-        print(f'Using ARN from existing role: {role_name}')
+        print(f"Using ARN from existing role: {role_name}")
         response = iam.get_role(RoleName=role_name)
-        return response['Role']['Arn']
+        return response["Role"]["Arn"]