From 75c8aeee5fb51abe377742ad51d66f13d52dd180 Mon Sep 17 00:00:00 2001 From: Andy Lee Date: Mon, 25 Aug 2025 15:48:04 -0700 Subject: [PATCH] style: format --- benchmarks/enron_emails/data/.gitignore | 2 +- benchmarks/enron_emails/evaluate_enron_emails.py | 6 +++++- .../leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py | 4 +++- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/benchmarks/enron_emails/data/.gitignore b/benchmarks/enron_emails/data/.gitignore index 361ae0f..c4ffae9 100644 --- a/benchmarks/enron_emails/data/.gitignore +++ b/benchmarks/enron_emails/data/.gitignore @@ -1 +1 @@ -downloads/ \ No newline at end of file +downloads/ diff --git a/benchmarks/enron_emails/evaluate_enron_emails.py b/benchmarks/enron_emails/evaluate_enron_emails.py index d780c59..e8c3db0 100644 --- a/benchmarks/enron_emails/evaluate_enron_emails.py +++ b/benchmarks/enron_emails/evaluate_enron_emails.py @@ -416,6 +416,7 @@ def main(): # Build non-compact index for fast binary search (recompute_embeddings=False) from pathlib import Path + index_path = Path(args.index) non_compact_index_path = str(index_path.parent / f"{index_path.stem}_noncompact.leann") enron_eval.create_non_compact_index_for_comparison(non_compact_index_path) @@ -428,6 +429,7 @@ def main(): results_out["stage3"] = sweep # Persist default stage 3 results near the index for Stage 4 auto-pickup from pathlib import Path + default_stage3_path = Path(args.index).parent / "enron_stage3_results.json" with open(default_stage3_path, "w", encoding="utf-8") as f: json.dump({"stage3": sweep}, f, indent=2) @@ -476,7 +478,9 @@ def main(): prev = json.load(f) complexity = prev.get("stage3", {}).get("best_complexity") if complexity is None: - raise SystemExit("❌ Stage 4: No --complexity and no best_complexity found in saved Stage 3 results") + raise SystemExit( + "❌ Stage 4: No --complexity and no best_complexity found in saved Stage 3 results" + ) print(f"📥 Using best complexity from saved Stage 3: {complexity}") else: raise SystemExit( diff --git a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py index 4af18e2..38fe599 100644 --- a/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py +++ b/packages/leann-backend-hnsw/leann_backend_hnsw/hnsw_backend.py @@ -274,7 +274,9 @@ class HNSWSearcher(BaseSearcher): return self._id_map[x] return str(x) - string_labels = [[map_label(int(l)) for l in batch_labels] for batch_labels in labels] + string_labels = [ + [map_label(int(label)) for label in batch_labels] for batch_labels in labels + ] else: string_labels = [ [str(int_label) for int_label in batch_labels] for batch_labels in labels