zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-08 12:24:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make RAPTOR GMM robust on small reduced clusters (#16632)

Browse Source
This commit is contained in:
Mattie Schraeder
2026-07-06 08:09:35 -05:00
committed by GitHub
parent 85b565244d
commit 8a19c6aa5a
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/unit_test/rag/test_raptor_psi_tree_builder.py
View File

@@ -245,7 +245,7 @@ def test_get_optimal_clusters_evaluates_upper_bound_candidate(monkeypatch, rapto
evaluated = []
class RecordingGaussianMixture:
def __init__(self, n_components, random_state=None):
def __init__(self, n_components, random_state=None, **kwargs):
self.n_components = n_components
evaluated.append(n_components)