Fix Redshift base provisioning value

mitdbg · Dec 18, 2023 · c1b823e · c1b823e
1 parent 429b402
commit c1b823e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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()
+)