From c1b823e38271d8c3af75cea56628e39beb12e285 Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Mon, 18 Dec 2023 16:42:29 -0500 Subject: [PATCH] Fix Redshift base provisioning value --- src/brad/planner/scoring/performance/unified_redshift.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/brad/planner/scoring/performance/unified_redshift.py b/src/brad/planner/scoring/performance/unified_redshift.py index 944650ee..2fb6c592 100644 --- a/src/brad/planner/scoring/performance/unified_redshift.py +++ b/src/brad/planner/scoring/performance/unified_redshift.py @@ -253,4 +253,7 @@ def add_debug_values(self, dest: Dict[str, int | float | str]) -> None: dest.update(self.debug_values) -_REDSHIFT_BASE_RESOURCE_VALUE = redshift_num_cpus(Provisioning("dc2.large", 2)) +_REDSHIFT_BASE_PROV = Provisioning("dc2.large", 2) +_REDSHIFT_BASE_RESOURCE_VALUE = ( + redshift_num_cpus(_REDSHIFT_BASE_PROV) * _REDSHIFT_BASE_PROV.num_nodes() +)