diff --git a/src/brad/planner/scoring/data_access/precomputed_values.py b/src/brad/planner/scoring/data_access/precomputed_values.py index 19e3306e..6f536fc0 100644 --- a/src/brad/planner/scoring/data_access/precomputed_values.py +++ b/src/brad/planner/scoring/data_access/precomputed_values.py @@ -145,11 +145,15 @@ def apply_access_statistics(self, workload: Workload) -> None: if "<=>" in q.raw_query: special_vector_queries.append(wqi) applied_athena[special_vector_queries] = 0.0 + applied_aurora[special_vector_queries] = 0.0 # Check for unmatched queries. num_unmatched_athena = np.isnan(applied_athena).sum() if num_unmatched_athena > 0: - raise RuntimeError("Unmatched queries: " + num_unmatched_athena) + raise RuntimeError("Unmatched Athena queries: " + num_unmatched_athena) + num_unmatched_aurora = np.isnan(applied_aurora).sum() + if num_unmatched_aurora > 0: + raise RuntimeError("Unmatched Aurora queries: " + num_unmatched_aurora) workload.set_predicted_data_access_statistics( aurora_pages=applied_aurora,