diff --git a/rag/nlp/__init__.py b/rag/nlp/__init__.py index 2aaf98aa..eb529f68 100644 --- a/rag/nlp/__init__.py +++ b/rag/nlp/__init__.py @@ -466,7 +466,7 @@ def hierarchical_merge(bull, sections, depth): jj = binary_search(levels[ii], j) if jj < 0: continue - if jj > cks[-1][-1]: + if levels[ii][jj] > cks[-1][-1]: cks[-1].pop(-1) cks[-1].append(levels[ii][jj]) for ii in cks[-1]: