Fix resolution for role names with paths

Author: Jaipreet Singh

sagemaker_studio_image_build/cli.py

@@ -37,11 +37,12 @@ def get_role(args):
         import sagemaker
 
         logging.basicConfig(level=logging.INFO)
-
-        return sagemaker.get_execution_role().split("/")[-1]
+        # arn:aws:iam::$account_id:role/$path/$name -> $path/$name
+        return "/".join(sagemaker.get_execution_role().split(":")[-1].split("/")[1:])
     except ValueError as e:
         raise ValueError(
-            "Unable to determine execution role. Please provide via --role", e
+            "Unable to determine execution role. Please provide via the --role argument",
+            e,
         )
 
 

setup.py

@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="sagemaker_studio_image_build",
-    version="0.3",
+    version="0.3.1",
     author="Amazon Web Services",
     description="Build Docker Images in Amazon SageMaker Studio using AWS CodeBuild",
     long_description=long_description,