Skip to content

Commit 2d52a09

Browse files
committed
fix(bench): remove references to deleted Python diffctx modules
1 parent 213279b commit 2d52a09

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

benchmarks/contextbench_diffctx.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -169,14 +169,11 @@ def run_baseline_bm25(repo_dir: Path, budget: int = 8000) -> dict | None:
169169

170170

171171
def _get_diffctx_config() -> dict:
172-
from treemapper.diffctx.config.limits import UTILITY as _UTILITY
173-
from treemapper.diffctx.filtering import _LOW_RELEVANCE_THRESHOLD, _MAX_CONTEXT_FRAGMENTS_PER_FILE
174-
175172
return {
176-
"low_relevance_threshold": _LOW_RELEVANCE_THRESHOLD,
177-
"proximity_decay": _UTILITY.proximity_decay,
178-
"peripheral_cap": _UTILITY.peripheral_cap,
179-
"max_context_frags_per_file": _MAX_CONTEXT_FRAGMENTS_PER_FILE,
173+
"low_relevance_threshold": 0.015,
174+
"proximity_decay": 0.30,
175+
"peripheral_cap": 0.15,
176+
"max_context_frags_per_file": 30,
180177
}
181178

182179

@@ -452,9 +449,7 @@ def _print_cross_seed_summary(all_seed_results: dict[int, list[dict]]) -> None:
452449
def main():
453450
import argparse
454451

455-
from treemapper.diffctx.filtering import _LOW_RELEVANCE_THRESHOLD
456-
457-
print(f"diffctx _LOW_RELEVANCE_THRESHOLD = {_LOW_RELEVANCE_THRESHOLD}", file=sys.stderr)
452+
print("diffctx backend: Rust (_diffctx)", file=sys.stderr)
458453

459454
parser = argparse.ArgumentParser()
460455
parser.add_argument("--limit", type=int, default=3)

0 commit comments

Comments
 (0)