Revert "cleanup CI fail"

This reverts commit 6dc844e.

Author: KRRT7

codeflash/benchmarking/function_ranker.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from pathlib import Path
 from typing import TYPE_CHECKING
 
 from codeflash.cli_cmds.console import console, logger
@@ -8,8 +9,6 @@
 from codeflash.tracing.profile_stats import ProfileStats
 
 if TYPE_CHECKING:
-    from pathlib import Path
-
     from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 
 
@@ -82,14 +81,47 @@ def load_function_stats(self) -> None:
             self._function_stats = {}
 
     def _get_function_stats(self, function_to_optimize: FunctionToOptimize) -> dict | None:
-        file_path = function_to_optimize.file_path.as_posix()
-        key1 = f"{file_path}:{function_to_optimize.qualified_name}"
-        key2 = f"{file_path}:{function_to_optimize.function_name}"
-
-        stats = self._function_stats.get(key1)
-        if stats is not None:
-            return stats
-        return self._function_stats.get(key2)
+        # Get all possible path variations
+        original_path = function_to_optimize.file_path.as_posix()
+        resolved_path = function_to_optimize.file_path.resolve().as_posix()
+
+        # Try to get relative path from current working directory
+        try:
+            cwd_relative = function_to_optimize.file_path.relative_to(Path.cwd()).as_posix()
+        except ValueError:
+            cwd_relative = original_path
+
+        # Try to get relative path from trace file directory
+        try:
+            trace_dir = self.trace_file_path.parent
+            trace_relative = function_to_optimize.file_path.relative_to(trace_dir).as_posix()
+        except ValueError:
+            trace_relative = original_path
+
+        # All possible path variants to try
+        path_variants = [original_path, resolved_path, cwd_relative, trace_relative]
+
+        for file_path in path_variants:
+            key1 = f"{file_path}:{function_to_optimize.qualified_name}"
+            key2 = f"{file_path}:{function_to_optimize.function_name}"
+
+            stats = self._function_stats.get(key1)
+            if stats is not None:
+                return stats
+            stats = self._function_stats.get(key2)
+            if stats is not None:
+                return stats
+
+        # Try fuzzy matching as last resort - match by function name and filename
+        target_filename = function_to_optimize.file_path.name
+        for key, stats in self._function_stats.items():
+            if stats.get("function_name") == function_to_optimize.function_name and (
+                key.endswith(f"/{target_filename}") or target_filename in key
+            ):
+                return stats
+
+        logger.debug(f"Could not find stats for function {function_to_optimize.function_name} in {original_path}")
+        return None
 
     def get_function_ttx_score(self, function_to_optimize: FunctionToOptimize) -> float:
         stats = self._get_function_stats(function_to_optimize)

tests/test_function_ranker.py

@@ -14,7 +14,7 @@ def trace_file():
 
 @pytest.fixture
 def workload_functions():
-    workloads_file = (Path(__file__).parent.parent / "code_to_optimize/code_directories/simple_tracer_e2e/workload.py").resolve()
+    workloads_file = Path(__file__).parent.parent / "code_to_optimize/code_directories/simple_tracer_e2e/workload.py"
     functions_dict = find_all_functions_in_file(workloads_file)
     all_functions = []
     for functions_list in functions_dict.values():