go

Author: KRRT7

codeflash/code_utils/code_extractor.py

@@ -362,9 +362,10 @@ def add_needed_imports_from_module(
 
 
 def get_code(functions_to_optimize: list[FunctionToOptimize]) -> tuple[str | None, set[tuple[str, str]]]:
-    """Return the code for a function or methods in a Python module. functions_to_optimize is either a singleton
-    FunctionToOptimize instance, which represents either a function at the module level or a method of a class at the
-    module level, or it represents a list of methods of the same class.
+    """Return the code for a function or methods in a Python module.
+
+    functions_to_optimize is either a singleton FunctionToOptimize instance, which represents either a function at the
+    module level or a method of a class at the module level, or it represents a list of methods of the same class.
     """
     if (
         not functions_to_optimize

codeflash/code_utils/shell_utils.py

@@ -46,9 +46,8 @@ def save_api_key_to_rc(api_key) -> Result[str, str]:
     try:
         with open(shell_rc_path, "r+", encoding="utf8") as shell_file:
             shell_contents = shell_file.read()
-            if os.name == "nt":  # on Windows, we're writing a batch file
-                if not shell_contents:
-                    shell_contents = "@echo off"
+            if os.name == "nt" and not shell_contents:  # on windows we're writing to a batch file
+                shell_contents = "@echo off"
             existing_api_key = read_api_key_from_shell_config()
 
             if existing_api_key: