fix test

terrierteam · Dec 4, 2024 · 6efc78c · 6efc78c
1 parent acdd1e8
commit 6efc78c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/pyterrier_dr/flex/flatnav_retr.py b/pyterrier_dr/flex/flatnav_retr.py
@@ -37,7 +37,7 @@ def transform(self, inp: pd.DataFrame) -> pd.DataFrame:
             scores, dids = self.flatnav_index.search(
                 queries=query_vecs[qidx:qidx+QBATCH],
                 ef_search=self.ef_search,
-                K=self.num_results
+                K=min(self.num_results, len(self.flex_index)),
             )
             scores = -scores # distances -> scores
             for s, d in zip(scores, dids):

diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -2,7 +2,6 @@ pytest
 pytest-subtests
 pytest-cov
 pytest-json-report
-git+https://github.com/terrierteam/pyterrier_adaptive
 voyager
 FlagEmbedding
 faiss-cpu

diff --git a/requirements.txt b/requirements.txt
@@ -5,3 +5,4 @@ torch
 numpy>=1.21.0, <2.0.0
 npids
 sentence_transformers
+pyterrier-adaptive