Better support for popular SageMaker Studio images

Author: ivan-khvostishkov

kernel-lc-config.sh

@@ -16,16 +16,17 @@ VNC_PASSWORD="123456"
 # OR keep it as is and put the value into ~/.sm-ssh-owner to override
 LOCAL_USER_ID="AIDACKCEVSQ6C2EXAMPLE"
 
+set -e
 
 hostname
 cat /opt/ml/metadata/resource-metadata.json
 
-pip uninstall --root-user-action ignore -y -q awscli
-pip install --root-user-action ignore -q sagemaker-ssh-helper
+pip uninstall -y -q awscli
+pip install -q sagemaker-ssh-helper
 
 # Uncomment two lines below to update SageMaker SSH Helper to the latest dev version from main branch
-#git clone https://github.com/aws-samples/sagemaker-ssh-helper.git
-#cd sagemaker-ssh-helper && pip install . && cd ..
+#git clone https://github.com/aws-samples/sagemaker-ssh-helper.git ./sagemaker-ssh-helper/ || echo 'Already cloned'
+#cd ./sagemaker-ssh-helper/ && git pull --no-rebase && pip install . && cd ..
 
 apt-get -y update
 apt-get -y install procps

sagemaker_ssh_helper/cdk/iam_ssm/iam_ssm_stack_tests.py

@@ -70,6 +70,7 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
                                       "sagemaker:DescribeTrainingJob",
                                       "sagemaker:DescribeProcessingJob",
                                       "sagemaker:DescribeModel",
+                                      "sagemaker:DescribeEndpointConfig",
                                       "sagemaker:DescribeEndpoint",
                                       "sagemaker:DescribeTransformJob",
                                       "sagemaker:DescribeHyperParameterTuningJob",
@@ -78,6 +79,9 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
                                       "sagemaker:DeleteEndpoint",
                                       "sagemaker:InvokeEndpoint",
                                       "sagemaker:StopTrainingJob",
+                                      "sagemaker:CreateApp",
+                                      "sagemaker:DeleteApp",
+                                      "sagemaker:DescribeApp",
                                   ],
                                   resources=["*"]
                               ),