path normalization and tempdir fixes for windows

Author: KRRT7

codeflash/benchmarking/replay_test.py

@@ -162,7 +162,7 @@ def create_trace_replay_test_code(
                 benchmark_function_name=benchmark_function_name,
                 orig_function_name=function_name,
                 function_name=alias,
-                file_path=file_path,
+                file_path=Path(file_path).as_posix(),
                 max_run_count=max_run_count,
             )
         else:
@@ -176,7 +176,7 @@ def create_trace_replay_test_code(
                 test_body = test_class_method_body.format(
                     benchmark_function_name=benchmark_function_name,
                     orig_function_name=function_name,
-                    file_path=file_path,
+                    file_path=Path(file_path).as_posix(),
                     class_name_alias=class_name_alias,
                     class_name=class_name,
                     method_name=method_name,
@@ -187,7 +187,7 @@ def create_trace_replay_test_code(
                 test_body = test_static_method_body.format(
                     benchmark_function_name=benchmark_function_name,
                     orig_function_name=function_name,
-                    file_path=file_path,
+                    file_path=Path(file_path).as_posix(),
                     class_name_alias=class_name_alias,
                     class_name=class_name,
                     method_name=method_name,
@@ -198,7 +198,7 @@ def create_trace_replay_test_code(
                 test_body = test_method_body.format(
                     benchmark_function_name=benchmark_function_name,
                     orig_function_name=function_name,
-                    file_path=file_path,
+                    file_path=Path(file_path).as_posix(),
                     class_name_alias=class_name_alias,
                     class_name=class_name,
                     method_name=method_name,

codeflash/code_utils/code_utils.py

@@ -171,8 +171,9 @@ def get_run_tmp_file(file_path: Path) -> Path:
 
 
 def path_belongs_to_site_packages(file_path: Path) -> bool:
+    file_path_resolved = file_path.resolve()
     site_packages = [Path(p) for p in site.getsitepackages()]
-    return any(file_path.resolve().is_relative_to(site_package_path) for site_package_path in site_packages)
+    return any(file_path_resolved.is_relative_to(site_package_path) for site_package_path in site_packages)
 
 
 def is_class_defined_in_file(class_name: str, file_path: Path) -> bool:

codeflash/code_utils/coverage_utils.py

@@ -45,7 +45,7 @@ def generate_candidates(source_code_path: Path) -> list[str]:
     current_path = source_code_path.parent
 
     while current_path != current_path.parent:
-        candidate_path = str(Path(current_path.name) / candidates[-1])
+        candidate_path = (Path(current_path.name) / candidates[-1]).as_posix()
         candidates.append(candidate_path)
         current_path = current_path.parent
 

codeflash/code_utils/instrument_existing_tests.py

@@ -212,7 +212,9 @@ 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=f"{get_run_tmp_file(Path('test_return_values_')).as_posix()}"
+                                                ),
                                                 ast.FormattedValue(
                                                     value=ast.Name(id="codeflash_iteration", ctx=ast.Load()),
                                                     conversion=-1,

codeflash/models/models.py

@@ -147,7 +147,7 @@ def markdown(self) -> str:
         """Returns the markdown representation of the code, including the file path where possible."""
         return "\n".join(
             [
-                f"```python{':' + str(code_string.file_path) if code_string.file_path else ''}\n{code_string.code.strip()}\n```"
+                f"```python{':' + code_string.file_path.as_posix() if code_string.file_path else ''}\n{code_string.code.strip()}\n```"
                 for code_string in self.code_strings
             ]
         )

codeflash/result/create_pr.py

@@ -104,7 +104,7 @@ def existing_tests_source_for(
                 if greater:
                     rows.append(
                         [
-                            f"`{filename}::{qualified_name}`",
+                            f"`{filename.as_posix()}::{qualified_name}`",
                             f"{print_original_runtime}",
                             f"{print_optimized_runtime}",
                             f"⚠️{perf_gain}%",
@@ -113,7 +113,7 @@ def existing_tests_source_for(
                 else:
                     rows.append(
                         [
-                            f"`{filename}::{qualified_name}`",
+                            f"`{filename.as_posix()}::{qualified_name}`",
                             f"{print_original_runtime}",
                             f"{print_optimized_runtime}",
                             f"✅{perf_gain}%",

codeflash/verification/instrument_codeflash_capture.py

@@ -33,7 +33,7 @@ def instrument_codeflash_capture(
     modified_code = add_codeflash_capture_to_init(
         target_classes={class_parent.name},
         fto_name=function_to_optimize.function_name,
-        tmp_dir_path=str(get_run_tmp_file(Path("test_return_values"))),
+        tmp_dir_path=get_run_tmp_file(Path("test_return_values")).as_posix(),
         code=original_code,
         tests_root=tests_root,
         is_fto=True,
@@ -46,7 +46,7 @@ def instrument_codeflash_capture(
         modified_code = add_codeflash_capture_to_init(
             target_classes=helper_classes,
             fto_name=function_to_optimize.function_name,
-            tmp_dir_path=str(get_run_tmp_file(Path("test_return_values"))),
+            tmp_dir_path=get_run_tmp_file(Path("test_return_values")).as_posix(),
             code=original_code,
             tests_root=tests_root,
             is_fto=False,
@@ -124,7 +124,7 @@ def visit_ClassDef(self, node: ast.ClassDef) -> ast.ClassDef:
             keywords=[
                 ast.keyword(arg="function_name", value=ast.Constant(value=f"{node.name}.__init__")),
                 ast.keyword(arg="tmp_dir_path", value=ast.Constant(value=self.tmp_dir_path)),
-                ast.keyword(arg="tests_root", value=ast.Constant(value=str(self.tests_root))),
+                ast.keyword(arg="tests_root", value=ast.Constant(value=self.tests_root.as_posix())),
                 ast.keyword(arg="is_fto", value=ast.Constant(value=self.is_fto)),
             ],
         )