From 77483b1e58e1720bef7c201a0a898c037a05bded Mon Sep 17 00:00:00 2001 From: Stephen Hu <812791840@qq.com> Date: Tue, 17 Mar 2026 15:56:51 +0800 Subject: [PATCH] refactor: remove useless variable in raptor (#13648) ### What problem does this PR solve? remove useless variable in raptor ### Type of change - [x] Refactoring --- rag/raptor.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/rag/raptor.py b/rag/raptor.py index ac2325d64..72e941296 100644 --- a/rag/raptor.py +++ b/rag/raptor.py @@ -161,7 +161,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval: if self._error_count >= self._max_errors: raise RuntimeError(f"RAPTOR aborted after {self._error_count} errors. Last error: {exc}") from exc - labels = [] while end - start > 1: self._check_task_canceled(task_id, "layer processing") @@ -170,7 +169,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval: await summarize([start, start + 1]) if callback: callback(msg="Cluster one layer: {} -> {}".format(end - start, len(chunks) - end)) - labels.extend([0, 0]) layers.append((end, len(chunks))) start = end end = len(chunks) @@ -208,7 +206,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval: raise assert len(chunks) - end == n_clusters, "{} vs. {}".format(len(chunks) - end, n_clusters) - labels.extend(lbls) layers.append((end, len(chunks))) if callback: callback(msg="Cluster one layer: {} -> {}".format(end - start, len(chunks) - end))