stats

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,46 +82,16 @@ 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}")
+        logger.debug(
+            f"Could not find stats for function {function_to_optimize.function_name} in file {target_filename}"
+        )
         return None
 
     def get_function_ttx_score(self, function_to_optimize: FunctionToOptimize) -> float: