optionally set base_rpu

aws · Apr 18, 2024 · edbe617 · edbe617
1 parent 371a92e
commit edbe617
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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(