pre-commit

Author: KRRT7

codeflash/benchmarking/function_ranker.py

@@ -141,7 +141,7 @@ def get_function_stats_summary(self, function_to_optimize: FunctionToOptimize) -
 
     def get_function_addressable_time(self, function_to_optimize: FunctionToOptimize) -> float:
         """Get the addressable time in nanoseconds for a function.
-        
+
         Addressable time = own_time + (time_in_callees / call_count)
         This represents the runtime of the function plus runtime of immediate dependent functions.
         """

codeflash/optimization/optimizer.py

@@ -105,7 +105,9 @@ def run_benchmarks(
                         f"No valid benchmarks found in {self.args.benchmarks_root} for functions to optimize, continuing optimization"
                     )
                 else:
-                    function_benchmark_timings = CodeFlashBenchmarkPlugin.get_function_benchmark_timings(self.trace_file)
+                    function_benchmark_timings = CodeFlashBenchmarkPlugin.get_function_benchmark_timings(
+                        self.trace_file
+                    )
                     total_benchmark_timings = CodeFlashBenchmarkPlugin.get_benchmark_timings(self.trace_file)
                     function_to_results = validate_and_format_benchmark_table(
                         function_benchmark_timings, total_benchmark_timings