def main()

in src/preprocess/get_base_image.py [0:0]


def main(framework, region, version, instance_type):
    return sagemaker.image_uris.retrieve(
        framework,
        region,
        version,
        instance_type=instance_type,
        py_version="py3",
        image_scope="training",
    )