Handle empty slices

mitdbg · Dec 4, 2023 · 6f180d1 · 6f180d1
1 parent 908c91c
commit 6f180d1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/brad/planner/scoring/performance/unified_aurora.py b/src/brad/planner/scoring/performance/unified_aurora.py
@@ -230,6 +230,9 @@ def predict_query_latency_load_resources(
         cpu_util: float,
         ctx: "ScoringContext",
     ) -> npt.NDArray:
+        if base_predicted_latency.shape[0] == 0:
+            return base_predicted_latency
+
         # 1. Compute each query's expected run time on the given provisioning.
         resource_factor = _AURORA_BASE_RESOURCE_VALUE / aurora_num_cpus(to_prov)
         basis = np.array([resource_factor, 1.0])

diff --git a/src/brad/planner/scoring/performance/unified_redshift.py b/src/brad/planner/scoring/performance/unified_redshift.py
@@ -158,6 +158,9 @@ def predict_query_latency_load_resources(
         overall_cpu_denorm: float,
         ctx: "ScoringContext",
     ) -> npt.NDArray:
+        if base_predicted_latency.shape[0] == 0:
+            return base_predicted_latency
+
         # 1. Compute each query's expected run time on the given provisioning.
         resource_factor = _REDSHIFT_BASE_RESOURCE_VALUE / (
             redshift_num_cpus(to_prov) * to_prov.num_nodes()