pass

Author: KRRT7

codeflash/code_utils/code_utils.py

@@ -9,6 +9,7 @@
 
 from codeflash.cli_cmds.console import logger
 
+_tmpdir = TemporaryDirectory(prefix="codeflash_")
 
 def get_qualified_name(module_name: str, full_qualified_name: str) -> str:
     if not full_qualified_name:
@@ -78,10 +79,12 @@ def get_all_function_names(code: str) -> tuple[bool, list[str]]:
     return True, function_names
 
 
-def get_run_tmp_file(file_path: Path) -> Path:
+def get_run_tmp_file(file_path: Path | str) -> Path:
     if not hasattr(get_run_tmp_file, "tmpdir"):
-        get_run_tmp_file.tmpdir = TemporaryDirectory(prefix="codeflash_")
-    return Path(get_run_tmp_file.tmpdir.name) / file_path
+        get_run_tmp_file.tmpdir = _tmpdir
+    if isinstance(file_path, str):
+        file_path = Path(file_path)
+    return (Path(get_run_tmp_file.tmpdir.name) / file_path).resolve()
 
 
 def path_belongs_to_site_packages(file_path: Path) -> bool:

codeflash/code_utils/instrument_existing_tests.py

@@ -213,7 +213,7 @@ def visit_FunctionDef(self, node: ast.FunctionDef, test_class_name: str | None =
                                     args=[
                                         ast.JoinedStr(
                                             values=[
-                                                ast.Constant(value=f"{get_run_tmp_file(Path('test_return_values_'))}"),
+                                                ast.Constant(value=get_run_tmp_file("test_return_values_").as_posix()),
                                                 ast.FormattedValue(
                                                     value=ast.Name(id="codeflash_iteration", ctx=ast.Load()),
                                                     conversion=-1,

tests/test_get_helper_code.py

@@ -213,10 +213,12 @@ def __call__(self, *args: _P.args, **kwargs: _P.kwargs) -> _R:
             lifespan=self.__duration__,
         )
 '''
-    with tempfile.NamedTemporaryFile(mode="w") as f:
-        f.write(code)
-        f.flush()
-        file_path = Path(f.name).resolve()
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_file_path = Path(temp_dir) / "temp_code.py"
+        with open(temp_file_path, "w") as f:
+            f.write(code)
+            f.flush()
+        file_path = temp_file_path.resolve()
         project_root_path = file_path.parent.resolve()
         function_to_optimize = FunctionToOptimize(
             function_name="__call__",