def get_data_prep_container_uri()

in code/workflow/implementations/autopilot/bp_init_stage.py [0:0]

• 29 lines of code
• 2 McCabe index (conditional complexity)

    def get_data_prep_container_uri(self, region):

        image_name = "sagemaker-data-wrangler-container"
        tag = "1.3.0"
        account = {    
            "us-west-1" : "926135532090",
            "us-west-2" : "174368400705",
            "us-east-1" : "663277389841",
            "us-east-2" : "415577184552",
            "ap-east-1" : "707077482487",
            "ap-northeast-1" : "649008135260",
            "ap-northeast-2" : "131546521161",
            "ap-southeast-1" : "119527597002",
            "ap-southeast-2" : "422173101802",
            "ap-south-1" : "089933028263",
            "eu-west-1" : "245179582081",
            "eu-west-2" : "894491911112",
            "eu-west-3" : "807237891255",
            "eu-south-1": "488287956546",
            "eu-central-1" : "024640144536",
            "ca-central-1" : "557239378090",
            "af-south-1" : "143210264188",
            "sa-east-1" : "424196993095",
            "me-south-1" : "376037874950"
           }[region]

        if not account :
            raise Exception("No entry found. Export your flow manually from \
                            the Data Wrangler console and update the account \
                            mapping in this function.")

        return f"{account}.dkr.ecr.{region}.amazonaws.com/{image_name}:{tag}"