Fix format

mitdbg · Dec 3, 2023 · 7bb57b1 · 7bb57b1
1 parent f453a9c
commit 7bb57b1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git 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
@@ -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