emb

langchain-ai · Apr 11, 2024 · 0a22fd4 · 0a22fd4
1 parent 6c254b2
commit 0a22fd4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/langsmith/_internal/_embedding_distance.py b/python/langsmith/_internal/_embedding_distance.py
@@ -3,6 +3,7 @@
 import logging
 from typing import (
     TYPE_CHECKING,
+    Any,
     Callable,
     List,
     Literal,
@@ -21,7 +22,7 @@
 
 logger = logging.getLogger(__name__)
 
-Matrix = Union[List[List[float]], List[np.ndarray], np.ndarray]
+Matrix = Union[List[List[float]], List[Any], Any]
 
 
 def cosine_similarity(X: Matrix, Y: Matrix) -> np.ndarray: