pre-commit run

Author: KRRT7

codeflash/code_utils/edit_generated_tests.py

@@ -41,7 +41,7 @@ def add_runtime_comments_to_generated_tests(
     optimized_runtime_by_test = optimized_test_results.usable_runtime_data_by_test_case()
 
     class RuntimeCommentTransformer(cst.CSTTransformer):
-        def __init__(self):
+        def __init__(self) -> None:
             self.in_test_function = False
             self.current_test_name = None
 
@@ -60,7 +60,9 @@ def leave_FunctionDef(self, original_node: cst.FunctionDef, updated_node: cst.Fu
             return updated_node
 
         def leave_SimpleStatementLine(
-            self, original_node: cst.SimpleStatementLine, updated_node: cst.SimpleStatementLine
+            self,
+            original_node: cst.SimpleStatementLine,  # noqa: ARG002
+            updated_node: cst.SimpleStatementLine,
         ) -> cst.SimpleStatementLine:
             if not self.in_test_function or not self.current_test_name:
                 return updated_node
@@ -69,14 +71,13 @@ def leave_SimpleStatementLine(
             # Handle both single statements and multiple statements on one line
             codeflash_assignment_found = False
             for stmt in updated_node.body:
-                if isinstance(stmt, cst.Assign):
-                    if (
-                        len(stmt.targets) == 1
-                        and isinstance(stmt.targets[0].target, cst.Name)
-                        and stmt.targets[0].target.value == "codeflash_output"
-                    ):
-                        codeflash_assignment_found = True
-                        break
+                if isinstance(stmt, cst.Assign) and (
+                    len(stmt.targets) == 1
+                    and isinstance(stmt.targets[0].target, cst.Name)
+                    and stmt.targets[0].target.value == "codeflash_output"
+                ):
+                    codeflash_assignment_found = True
+                    break
 
             if codeflash_assignment_found:
                 # Find matching test cases by looking for this test function name in the test results

codeflash/code_utils/time_utils.py

@@ -66,25 +66,31 @@ def format_with_precision(value: float, unit: str) -> str:
             return f"{value:.1f}{unit}"
         return f"{value:.2f}{unit}"
 
+    result = ""
     if nanoseconds < 1_000:
-        return f"{nanoseconds}ns"
-    if nanoseconds < 1_000_000:
+        result = f"{nanoseconds}ns"
+    elif nanoseconds < 1_000_000:
         # Convert to microseconds
         microseconds_int = nanoseconds // 1_000
         if count_significant_digits(microseconds_int) >= 3:
-            return f"{microseconds_int}μs"
-        microseconds_float = nanoseconds / 1_000
-        return format_with_precision(microseconds_float, "μs")
-    if nanoseconds < 1_000_000_000:
+            result = f"{microseconds_int}μs"
+        else:
+            microseconds_float = nanoseconds / 1_000
+            result = format_with_precision(microseconds_float, "μs")
+    elif nanoseconds < 1_000_000_000:
         # Convert to milliseconds
         milliseconds_int = nanoseconds // 1_000_000
         if count_significant_digits(milliseconds_int) >= 3:
-            return f"{milliseconds_int}ms"
-        milliseconds_float = nanoseconds / 1_000_000
-        return format_with_precision(milliseconds_float, "ms")
-    # Convert to seconds
-    seconds_int = nanoseconds // 1_000_000_000
-    if count_significant_digits(seconds_int) >= 3:
-        return f"{seconds_int}s"
-    seconds_float = nanoseconds / 1_000_000_000
-    return format_with_precision(seconds_float, "s")
+            result = f"{milliseconds_int}ms"
+        else:
+            milliseconds_float = nanoseconds / 1_000_000
+            result = format_with_precision(milliseconds_float, "ms")
+    else:
+        # Convert to seconds
+        seconds_int = nanoseconds // 1_000_000_000
+        if count_significant_digits(seconds_int) >= 3:
+            result = f"{seconds_int}s"
+        else:
+            seconds_float = nanoseconds / 1_000_000_000
+            result = format_with_precision(seconds_float, "s")
+    return result

codeflash/context/code_context_extractor.py

@@ -386,7 +386,7 @@ def get_function_to_optimize_as_function_source(
                     source_code=name.get_line_code(),
                     jedi_definition=name,
                 )
-        except Exception as e:  # noqa: PERF203
+        except Exception as e:
             logger.exception(f"Error while getting function source: {e}")
             continue
     raise ValueError(

pyproject.toml

@@ -191,7 +191,8 @@ ignore = [
     "T201",
     "PGH004",
     "S301",
-    "D104"
+    "D104",
+    "PERF203"
 ]
 
 [tool.ruff.lint.flake8-type-checking]