linting

Author: hinriksnaer

helion/autotuner/base_search.py

@@ -794,7 +794,9 @@ def parallel_overlapped_benchmark(
 
             next(iterator, None)  # Update progress bar
 
-        ordered_results: list[BenchmarkResult] = [results[i] for i in range(len(configs))]
+        ordered_results: list[BenchmarkResult] = [
+            results[i] for i in range(len(configs))
+        ]
 
         return ordered_results
 
@@ -1244,12 +1246,16 @@ def check_benchmark_stability(self) -> None:
         overlap_stability_threshold, overlap is automatically disabled.
         """
         rebenchmarked = [
-            m for m in self.population
+            m
+            for m in self.population
             if len(m.perfs) > 1 and min(m.perfs[0], m.perfs[1]) > 0
         ]
         if len(rebenchmarked) < 2:
             return
-        ratios = [max(m.perfs[0], m.perfs[1]) / min(m.perfs[0], m.perfs[1]) for m in rebenchmarked]
+        ratios = [
+            max(m.perfs[0], m.perfs[1]) / min(m.perfs[0], m.perfs[1])
+            for m in rebenchmarked
+        ]
         avg_ratio = sum(ratios) / len(ratios)
         threshold = self.settings.overlap_stability_threshold
         self.log(

helion/autotuner/surrogate_pattern_search.py

@@ -354,7 +354,9 @@ def _autotune(self) -> Config:
             self.rebenchmark(successful, desc="Verifying initial results")
             self.check_benchmark_stability()
         else:
-            self.rebenchmark_population(self.population, desc="Verifying initial results")
+            self.rebenchmark_population(
+                self.population, desc="Verifying initial results"
+            )
         self.population.sort(key=performance)
         starting_points = []
         for member in self.population[: self.copies]: