merge

langchain-ai · Apr 11, 2024 · acad86e · acad86e
1 parent 7e57a1a
commit acad86e
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 65 deletions.
diff --git a/python/langsmith/_expect.py b/python/langsmith/_expect.py
@@ -245,8 +245,11 @@ def embedding_distance(
 
 
         Examples:
-            >>> expect.embedding_distance("hello", "hi").to_be_less_than(0.5)
-        """
+            >>> expect.embedding_distance(
+            ...     prediction="hello",
+            ...     reference="hi",
+            ... ).to_be_less_than(0.5)
+        """  # noqa: E501
         from langsmith._internal._embedding_distance import EmbeddingDistance
 
         config = config or {}

diff --git a/python/langsmith/_internal/_edit_distance.py b/python/langsmith/_internal/_edit_distance.py
@@ -36,7 +36,9 @@ def evaluate(
     @staticmethod
     def _get_metric(distance: str, normalize_score: bool = False) -> Callable:
         try:
-            from rapidfuzz import distance as rf_distance
+            from rapidfuzz import (
+                distance as rf_distance,  # type: ignore[import-not-found]
+            )
         except ImportError:
             raise ImportError(
                 "This operation requires the rapidfuzz library to use."

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,
@@ -19,9 +20,7 @@
 
 logger = logging.getLogger(__name__)
 
-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:

diff --git a/python/tests/integration_tests/test__testing.py b/python/tests/integration_tests/test__testing.py