From 5e681c1d2fbc16a15602b9da0e587d995fe2f951 Mon Sep 17 00:00:00 2001
From: aldengolab <agolab@chanzuckerberg.com>
Date: Thu, 18 Jan 2024 13:13:44 -0800
Subject: [PATCH] chore: add account-level groups to policies

---
 databricks-default-cluster-policies/main.tf | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/databricks-default-cluster-policies/main.tf b/databricks-default-cluster-policies/main.tf
index 5cb5defd..36999cbe 100644
--- a/databricks-default-cluster-policies/main.tf
+++ b/databricks-default-cluster-policies/main.tf
@@ -135,6 +135,7 @@ module "large_personal_compute_cluster_policy" {
       "hidden" : false
     },
   })
+  grantees = ["CZI - Large Personal Compute"]
 }
 
 module "power_user_compute_cluster_policy" {
@@ -166,7 +167,7 @@ module "power_user_compute_cluster_policy" {
     },
   })
 
-  grantees = [local.power_user_group_name]
+  grantees = [local.power_user_group_name, "CZI - Power User Compute"]
 }
 module "job_compute_cluster_policy" {
   source = "../databricks-cluster-policy"
@@ -178,7 +179,7 @@ module "job_compute_cluster_policy" {
 
   policy_overrides = local.logging_override
 
-  grantees = [local.power_user_group_name]
+  grantees = [local.power_user_group_name, "CZI - Job Compute"]
 }
 
 module "small_job_compute_cluster_policy" {
@@ -197,7 +198,7 @@ module "small_job_compute_cluster_policy" {
     }
   })
 
-  grantees = [local.power_user_group_name]
+  grantees = [local.power_user_group_name, "CZI - Small Job Compute"]
 }
 
 ## Fully custom policies
@@ -246,6 +247,7 @@ module "large_gpu_large_clusters_cluster_policy" {
       "defaultValue" : "g4dn.xlarge"
     },
   })
+  grantees = ["CZI - Large GPU Large Clusters"]
 }
 
 module "large_gpu_personal_cluster_policy" {
@@ -281,6 +283,7 @@ module "large_gpu_personal_cluster_policy" {
       "defaultValue" : "g4dn.xlarge"
     },
   })
+  grantees = ["CZI - Large GPU Personal"]
 }
 
 module "large_gpu_small_clusters_cluster_policy" {
@@ -328,6 +331,7 @@ module "large_gpu_small_clusters_cluster_policy" {
       "defaultValue" : "g4dn.xlarge"
     },
   })
+  grantees = ["CZI - Large GPU Small Clusters"]
 }
 
 module "small_clusters" {
@@ -380,6 +384,7 @@ module "small_clusters" {
       "hidden" : false
     },
   })
+  grantees = ["CZI - Small Clusters"]
 }
 
 module "superset_compute_cluster_policy" {
@@ -415,4 +420,5 @@ module "superset_compute_cluster_policy" {
       "defaultValue" : "superset_pool"
     },
   })
+  grantees = ["CZI - Superset Compute"]
 }