github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-10-18 23:50:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Add ruff rule PGH003 (#30812)

Browse Source 
See https://docs.astral.sh/ruff/rules/blanket-type-ignore/

---------

Co-authored-by: Chester Curme <chester.curme@gmail.com>
This commit is contained in:
Christophe Bornet
2025-04-14 04:32:13 +02:00
committed by GitHub
parent f005988e31
commit ada740b5b9
148 changed files with 448 additions and 419 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
libs/community/tests/integration_tests/retrievers/docarray/fixtures.py
View File

@@ -40,8 +40,8 @@ def init_weaviate() -> Generator[
# When initializing the Weaviate index, denote the field
# you want to search on with `is_embedding=True`
title: str
title_embedding: NdArray[32] = Field(is_embedding=True) # type: ignore
other_emb: NdArray[32] # type: ignore
title_embedding: NdArray[32] = Field(is_embedding=True) # type: ignore[call-overload]
other_emb: NdArray[32]
year: int
embeddings = FakeEmbeddings(size=32)
@@ -87,8 +87,8 @@ def init_elastic() -> Generator[
class MyDoc(BaseDoc):
title: str
title_embedding: NdArray[32] # type: ignore
other_emb: NdArray[32] # type: ignore
title_embedding: NdArray[32]
other_emb: NdArray[32]
year: int
embeddings = FakeEmbeddings(size=32)
@@ -124,8 +124,8 @@ def init_qdrant() -> Tuple[QdrantDocumentIndex, rest.Filter, FakeEmbeddings]:
class MyDoc(BaseDoc):
title: str
title_embedding: NdArray[32] # type: ignore
other_emb: NdArray[32] # type: ignore
title_embedding: NdArray[32]
other_emb: NdArray[32]
year: int
embeddings = FakeEmbeddings(size=32)
@@ -168,8 +168,8 @@ def init_in_memory() -> Tuple[InMemoryExactNNIndex, Dict[str, Any], FakeEmbeddin
class MyDoc(BaseDoc):
title: str
title_embedding: NdArray[32] # type: ignore
other_emb: NdArray[32] # type: ignore
title_embedding: NdArray[32]
other_emb: NdArray[32]
year: int
embeddings = FakeEmbeddings(size=32)
@@ -205,8 +205,8 @@ def init_hnsw(
class MyDoc(BaseDoc):
title: str
title_embedding: NdArray[32] # type: ignore
other_emb: NdArray[32] # type: ignore
title_embedding: NdArray[32]
other_emb: NdArray[32]
year: int
embeddings = FakeEmbeddings(size=32)

20
libs/community/tests/integration_tests/retrievers/test_zep.py
View File

@@ -61,12 +61,8 @@ def zep_retriever(
mock_zep_client.memory = mocker.patch(
"zep_python.memory.client.MemoryClient", autospec=True
)
mock_zep_client.memory.search_memory.return_value = copy.deepcopy( # type: ignore
search_results
)
mock_zep_client.memory.asearch_memory.return_value = copy.deepcopy( # type: ignore
search_results
)
mock_zep_client.memory.search_memory.return_value = copy.deepcopy(search_results)
mock_zep_client.memory.asearch_memory.return_value = copy.deepcopy(search_results)
zep = ZepRetriever(session_id="123", url="http://localhost:8000") # type: ignore[call-arg]
zep.zep_client = mock_zep_client
return zep
@@ -93,13 +89,7 @@ def _test_documents(
) -> None:
assert len(documents) == 2
for i, document in enumerate(documents):
assert document.page_content == search_results[i].message.get( # type: ignore
"content"
)
assert document.metadata.get("uuid") == search_results[i].message.get( # type: ignore
"uuid"
)
assert document.metadata.get("role") == search_results[i].message.get( # type: ignore
"role"
)
assert document.page_content == search_results[i].message.get("content")
assert document.metadata.get("uuid") == search_results[i].message.get("uuid")
assert document.metadata.get("role") == search_results[i].message.get("role")
assert document.metadata.get("score") == search_results[i].dist