From 7bb57b135e7ef3c52db9c255b6ad0aadba087944 Mon Sep 17 00:00:00 2001 From: Geoffrey Yu Date: Sun, 3 Dec 2023 17:49:46 -0500 Subject: [PATCH] Fix format --- src/brad/config/planner.py | 4 +++- src/brad/planner/scoring/performance/unified_redshift.py | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/brad/config/planner.py b/src/brad/config/planner.py index 76b81c97..f65adf4c 100644 --- a/src/brad/config/planner.py +++ b/src/brad/config/planner.py @@ -238,7 +238,9 @@ def redshift_scaling_coefs(self) -> npt.NDArray: def redshift_new_scaling_coefs(self) -> npt.NDArray: if self._redshift_new_scaling_coefs is None: coefs = self._raw["redshift_scaling_new"] - self._redshift_new_scaling_coefs = np.array([coefs["coef1"], coefs["coef2"]]) + self._redshift_new_scaling_coefs = np.array( + [coefs["coef1"], coefs["coef2"]] + ) return self._redshift_new_scaling_coefs def redshift_new_scaling_alpha(self) -> float: diff --git a/src/brad/planner/scoring/performance/unified_redshift.py b/src/brad/planner/scoring/performance/unified_redshift.py index 2f2144e9..0e57f26e 100644 --- a/src/brad/planner/scoring/performance/unified_redshift.py +++ b/src/brad/planner/scoring/performance/unified_redshift.py @@ -172,11 +172,15 @@ def predict_query_latency_load_resources( # 2. Compute the impact of system load. mean_service_time = alone_predicted_latency.mean() - cpu_util = overall_cpu_denorm / (redshift_num_cpus(to_prov) * to_prov.num_nodes()) + cpu_util = overall_cpu_denorm / ( + redshift_num_cpus(to_prov) * to_prov.num_nodes() + ) denom = max(1e-3, 1.0 - cpu_util) # Want to avoid division by 0. wait_sf = cpu_util / denom mean_wait_time = ( - mean_service_time * wait_sf * ctx.planner_config.redshift_new_scaling_alpha() + mean_service_time + * wait_sf + * ctx.planner_config.redshift_new_scaling_alpha() ) # Predicted running time is the query's execution time alone plus the