diff --git a/elasticsearch/helpers/vectorstore/_utils.py b/elasticsearch/helpers/vectorstore/_utils.py index df91b5cc9..d8f7b339f 100644 --- a/elasticsearch/helpers/vectorstore/_utils.py +++ b/elasticsearch/helpers/vectorstore/_utils.py @@ -100,7 +100,8 @@ def _cosine_similarity(X: Matrix, Y: Matrix) -> "npt.NDArray[np.float64]": X = np.array(X, dtype=np.float32) Y = np.array(Y, dtype=np.float32) - Z = 1 - np.array(simd.cdist(X, Y, metric="cosine")) + # https://github.com/ashvardanian/SimSIMD/issues/171 + Z = 1 - np.array(simd.cdist(X, Y, metric="cosine")) # type: ignore[arg-type] if isinstance(Z, float): return np.array([Z]) return np.array(Z) diff --git a/pyproject.toml b/pyproject.toml index ba83d6329..1a2563680 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -76,7 +76,7 @@ dev = [ ] docs = [ "sphinx", - "sphinx-rtd-theme", + "sphinx-rtd-theme>=2.0", "sphinx-autodoc-typehints" ]