From 0a22fd468eb70e0e9021e582d80f3a1cbf29440a Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Wed, 10 Apr 2024 18:44:47 -0700
Subject: [PATCH] emb

---
 python/langsmith/_internal/_embedding_distance.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/langsmith/_internal/_embedding_distance.py b/python/langsmith/_internal/_embedding_distance.py
index 3e6efb8b7..b11963b96 100644
--- 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: