diff --git a/profiling/graph_pydantic_conversion/benchmark_function.py b/profiling/graph_pydantic_conversion/benchmark_function.py index d80bafab1..413d0cde4 100644 --- a/profiling/graph_pydantic_conversion/benchmark_function.py +++ b/profiling/graph_pydantic_conversion/benchmark_function.py @@ -27,7 +27,7 @@ def benchmark_function(func: Callable, *args, num_runs: int = 5) -> Dict[str, An for _ in range(num_runs): # Start memory tracking tracemalloc.start() - initial_memory = process.memory_info().rss + # Measure execution time and CPU usage start_time = time.perf_counter() @@ -44,7 +44,7 @@ def benchmark_function(func: Callable, *args, num_runs: int = 5) -> Dict[str, An start_cpu_time.user + start_cpu_time.system ) current, peak = tracemalloc.get_traced_memory() - final_memory = process.memory_info().rss + # Store results diff --git a/pyproject.toml b/pyproject.toml index 8550a6edc..6a43bb603 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -116,6 +116,11 @@ mkdocstrings = {extras = ["python"], version = "^0.26.2"} [tool.ruff] # https://beta.ruff.rs/docs/ line-length = 100 +exclude = [ + "migrations/", # Ignore migrations directory + "notebooks/", # Ignore notebook files + "build/", # Ignore build directory +] [tool.ruff.lint] ignore = ["F401"]