diff --git a/common/util.py b/common/util.py index 007afcb0..e0f308b6 100644 --- 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(