From edbe6172e880453f460862d32ccbd1107329cdc6 Mon Sep 17 00:00:00 2001
From: Jessie Chen <jiezhec@amazon.com>
Date: Thu, 18 Apr 2024 13:16:18 -0700
Subject: [PATCH] optionally set base_rpu

---
 common/util.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/common/util.py b/common/util.py
index e0f308b6..192542cf 100644
--- a/common/util.py
+++ b/common/util.py
@@ -91,6 +91,8 @@ def cluster_dict(endpoint, is_serverless=False, start_time=None, end_time=None):
             cluster["instance"] = "Serverless"
             if response.get("workgroup").get("baseCapacity"):
                 cluster["base_rpu"] = response.get("workgroup").get("baseCapacity")
+            else:
+                cluster["base_rpu"] = "N/A"
         except Exception as e:
             if e.response["Error"]["Code"] == "ResourceNotFoundException":
                 logger.warning(