def get_ami_id()

in aws/autoscaling.py [0:0]

• 20 lines of code
• 1 McCabe index (conditional complexity)

    def get_ami_id(self, accelerator):
        """
        Use EKS optimized AMI since it has everything we need pre-installed
        """

        eks_owner_id = "602401143452"
        eks_amis = {
            Accelerator.NONE: "amazon-eks-node-1.14-v20190927",
            Accelerator.GPU: "amazon-eks-gpu-node-1.14-v20190927",
        }

        res = self._ec2.describe_images(
            Filters=[
                {"Name": "owner-id", "Values": [eks_owner_id]},
                {
                    "Name": "name",
                    "Values": [eks_amis.get(accelerator, Accelerator.NONE)],
                },
            ]
        )
        images = res["Images"]
        assert (
            len(images) == 1
        ), f"Multiple EKS AMIs found for {self._session.aws_region()}"
        return images[0]["ImageId"]