type ignores for mocking methods

langchain-ai · May 3, 2024 · e41dca1 · e41dca1
1 parent dedc31b
commit e41dca1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/libs/elasticsearch/tests/unit_tests/test_vectorstores.py b/libs/elasticsearch/tests/unit_tests/test_vectorstores.py
@@ -193,7 +193,7 @@ def test_agent_header(self, store: ElasticsearchStore) -> None:
     def test_similarity_search(
         self, store: ElasticsearchStore, static_hits: List[Dict]
     ) -> None:
-        store._store.search = Mock(return_value=static_hits)
+        store._store.search = Mock(return_value=static_hits)  # type: ignore[assignment]
         actual1 = store.similarity_search(
             query="test",
             k=7,
@@ -210,7 +210,7 @@ def test_similarity_search(
             custom_query=self.dummy_custom_query,
         )
 
-        store._store.search = Mock(return_value=static_hits)
+        store._store.search = Mock(return_value=static_hits)  # type: ignore[assignment]
 
         actual2 = store.similarity_search_with_score(
             query="test",
@@ -230,7 +230,7 @@ def test_similarity_search(
     def test_similarity_search_by_vector_with_relevance_scores(
         self, store: ElasticsearchStore, static_hits: List[Dict]
     ) -> None:
-        store._store.search = Mock(return_value=static_hits)
+        store._store.search = Mock(return_value=static_hits)  # type: ignore[assignment]
         actual = store.similarity_search_by_vector_with_relevance_scores(
             embedding=[1, 2, 3],
             k=7,
@@ -333,7 +333,7 @@ def test_max_marginal_relevance_search(
         embeddings: Embeddings,
         static_hits: List[Dict],
     ) -> None:
-        hybrid_store._store.max_marginal_relevance_search = Mock(
+        hybrid_store._store.max_marginal_relevance_search = Mock(  # type: ignore[assignment]
             return_value=static_hits
         )
         actual = hybrid_store.max_marginal_relevance_search(