optionally set base_rpu

aws · Apr 18, 2024 · 371a92e · 371a92e
1 parent 7a1a1ee
commit 371a92e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/util.py b/common/util.py
@@ -87,10 +87,10 @@ def cluster_dict(endpoint, is_serverless=False, start_time=None, end_time=None):
             response = aws_service_helper.redshift_get_serverless_workgroup(
                 workgroup_name, cluster.get("region")
             )
-
             cluster["num_nodes"] = "N/A"
             cluster["instance"] = "Serverless"
-            cluster["base_rpu"] = response["workgroup"]["baseCapacity"]
+            if response.get("workgroup").get("baseCapacity"):
+                cluster["base_rpu"] = response.get("workgroup").get("baseCapacity")
         except Exception as e:
             if e.response["Error"]["Code"] == "ResourceNotFoundException":
                 logger.warning(