From da0c2c99a096151bcc764d1ae3f7e707310f1e9d Mon Sep 17 00:00:00 2001 From: Patrick Ames Date: Mon, 15 Feb 2021 16:29:36 -0800 Subject: [PATCH] [autoscaler] Fix bad reference error when specifying IamInstanceProfile by name in config. (#14083) --- python/ray/autoscaler/_private/aws/config.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/ray/autoscaler/_private/aws/config.py b/python/ray/autoscaler/_private/aws/config.py index 2fb90787b..9aa3e6d85 100644 --- a/python/ray/autoscaler/_private/aws/config.py +++ b/python/ray/autoscaler/_private/aws/config.py @@ -155,11 +155,11 @@ def log_to_cli(config): _tags=workers_tags) tags = {"default": _log_info["head_instance_profile_src"] == "default"} - cli_logger.labeled_value( - "IAM Profile", - "{}", - _arn_to_name(config["head_node"]["IamInstanceProfile"]["Arn"]), - _tags=tags) + profile_arn = config["head_node"]["IamInstanceProfile"].get("Arn") + profile_name = _arn_to_name(profile_arn) \ + if profile_arn \ + else config["head_node"]["IamInstanceProfile"]["Name"] + cli_logger.labeled_value("IAM Profile", "{}", profile_name, _tags=tags) if ("KeyName" in config["head_node"] and "KeyName" in config["worker_nodes"]):