chore(ci): remove paru-bin submodule and config to fix checkout --recurse-submodules

This commit is contained in:
Andy Lee
2025-08-25 16:08:16 -07:00
parent 75c8aeee5f
commit ef4c69d128
4 changed files with 10 additions and 20 deletions

View File

@@ -1,19 +1,8 @@
#!/usr/bin/env python3
"""
Run DiskANN with real NQ queries (search-only timing).
Steps:
- Load queries from nq_open.jsonl
- Compute embeddings (facebook/contriever-msmarco) once upfront
- Search via DiskANN (no recompute, no batching), measure per-query latency
Example:
python benchmarks/bm25_diskann_baselines/run_diskann_nq.py \
--index-dir benchmarks/data/indices/diskann_rpj_wiki \
--index-prefix ann \
--queries-file benchmarks/data/queries/nq_open.jsonl \
--num-queries 200 --top-k 10 --complexity 120 --threads 1 --beam-width 1
"""
# /// script
# dependencies = [
# "leann-backend-diskann"
# ]
# ///
import argparse
import json
@@ -47,7 +36,7 @@ def main() -> None:
ap.add_argument("--queries-file", default="benchmarks/data/queries/nq_open.jsonl")
ap.add_argument("--num-queries", type=int, default=200)
ap.add_argument("--top-k", type=int, default=10)
ap.add_argument("--complexity", type=int, default=120)
ap.add_argument("--complexity", type=int, default=62)
ap.add_argument("--threads", type=int, default=1)
ap.add_argument("--beam-width", type=int, default=1)
ap.add_argument("--cache-mechanism", type=int, default=2)