cleanup CI fail

Author: KRRT7

codeflash/benchmarking/function_ranker.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-from pathlib import Path
 from typing import TYPE_CHECKING
 
 from codeflash.cli_cmds.console import console, logger
@@ -9,6 +8,8 @@
 from codeflash.tracing.profile_stats import ProfileStats
 
 if TYPE_CHECKING:
+    from pathlib import Path
+
     from codeflash.discovery.functions_to_optimize import FunctionToOptimize
 
 
@@ -81,47 +82,14 @@ def load_function_stats(self) -> None:
             self._function_stats = {}
 
     def _get_function_stats(self, function_to_optimize: FunctionToOptimize) -> dict | None:
-        # 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
+        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)
 
     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"
+    workloads_file = (Path(__file__).parent.parent / "code_to_optimize/code_directories/simple_tracer_e2e/workload.py").resolve()
     functions_dict = find_all_functions_in_file(workloads_file)
     all_functions = []
     for functions_list in functions_dict.values():