ragtools 0.2.0__tar.gz → 0.2.1__tar.gz

{ragtools-0.2.0 → ragtools-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ragtools
-Version: 0.2.0
+Version: 0.2.1
 Summary: Add your description here
 Author: Wannes Vantorre
 Author-email: Wannes Vantorre <vantorrewannes@gmail.com>

{ragtools-0.2.0 → ragtools-0.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ragtools"
-version = "0.2.0"
+version = "0.2.1"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.14"

{ragtools-0.2.0 → ragtools-0.2.1}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "ragtools"
-version = "0.2.0"
+version = "0.2.1"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{ragtools-0.2.0 → ragtools-0.2.1}/src/ragtools/indexes.py

@@ -138,11 +138,11 @@ class SparseEmbeddingIndex[K: Hashable]:
     def keys(self) -> builtins.set[K]:
         return builtins.set(self._row_of)
 
-    def closest(self, key: K, k: int) -> tuple[K, ...]:
+    def closest(self, key: K, k: int) -> list[K]:
         if key not in self._row_of:
             raise KeyError(key)
         if k <= 0 or len(self._rows) <= 1 or len(self._vocab) == 0:
-            return ()
+            return []
         r = self._row_of[key]
         start, end = int(self._indptr[r]), int(self._indptr[r + 1])
         q = np.zeros(len(self._vocab), dtype=np.float32)
@@ -153,7 +153,7 @@ class SparseEmbeddingIndex[K: Hashable]:
         k_eff = min(k, len(self._rows) - 1)
         top = np.argpartition(-scores, k_eff - 1)[:k_eff]
         top = top[np.argsort(-scores[top], kind="stable")]
-        return tuple(self._rows[i] for i in top if scores[i] != -np.inf)
+        return [self._rows[i] for i in top if scores[i] != -np.inf]
 
     @property
     def _matrix(self) -> csr_matrix: