Skip to content

Commit

Permalink
reuse exiting policy
Browse files Browse the repository at this point in the history
  • Loading branch information
suzhoum committed Jul 12, 2024
1 parent a73ccde commit 2fb8489
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions src/autogluon/cloud/backend/ray_aws_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@
create_iam_policy,
create_iam_role,
create_instance_profile,
delete_iam_policy,
get_policy,
replace_iam_policy_place_holder,
replace_trust_relationship_place_holder,
Expand Down Expand Up @@ -80,11 +79,10 @@ def _setup_role_and_permission(self):
)
create_iam_role(role_name=RAY_AWS_ROLE_NAME, trust_relationship=trust_relationship)
policy_arn = get_policy(policy_name=RAY_AWS_POLICY_NAME)
if policy_arn is not None:
delete_iam_policy(policy_arn=policy_arn)
policy_arn = create_iam_policy(policy_name=RAY_AWS_POLICY_NAME, policy=iam_policy)
attach_iam_policy(role_name=RAY_AWS_ROLE_NAME, policy_arn=policy_arn)
attach_iam_policy(role_name=RAY_AWS_ROLE_NAME, policy_arn=ECR_READ_ONLY)
if policy_arn is None:
policy_arn = create_iam_policy(policy_name=RAY_AWS_POLICY_NAME, policy=iam_policy)
attach_iam_policy(role_name=RAY_AWS_ROLE_NAME, policy_arn=policy_arn)
attach_iam_policy(role_name=RAY_AWS_ROLE_NAME, policy_arn=ECR_READ_ONLY)
instance_profile_arn = create_instance_profile(instance_profile_name=RAY_INSTANCE_PROFILE_NAME)
if instance_profile_arn is not None:
add_role_to_instance_profile(instance_profile_name=RAY_INSTANCE_PROFILE_NAME, role_name=RAY_AWS_ROLE_NAME)
Expand Down

0 comments on commit 2fb8489

Please sign in to comment.