type ignores for mocking methods

Author: maxjakob

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,
@@ -335,7 +335,7 @@ def test_max_marginal_relevance_search(
     ) -> None:
         hybrid_store._store.max_marginal_relevance_search = Mock(
             return_value=static_hits
-        )
+        )  # type: ignore[assignment]
         actual = hybrid_store.max_marginal_relevance_search(
             query="qqq",
             k=8,