ref: Fix some ruff preview rules (#5156)

Fix some ruff preview rules
This commit is contained in:
Christophe Bornet 2024-12-11 14:37:29 +01:00 committed by GitHub
commit 5ee995e89d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 26 additions and 39 deletions

View file

@ -215,10 +215,9 @@ def test_vector_store_rag_add(ingestion_graph: Graph, rag_graph: Graph):
f"Vertices mismatch: {len(ingestion_graph_copy.vertices)} "
f"!= {len(ingestion_graph.vertices)} + {len(rag_graph.vertices)}"
)
assert len(ingestion_graph_copy.edges) == len(ingestion_graph.edges) + len(rag_graph.edges), (
f"Edges mismatch: {len(ingestion_graph_copy.edges)} "
f"!= {len(ingestion_graph.edges)} + {len(rag_graph.edges)}"
)
assert len(ingestion_graph_copy.edges) == len(ingestion_graph.edges) + len(
rag_graph.edges
), f"Edges mismatch: {len(ingestion_graph_copy.edges)} != {len(ingestion_graph.edges)} + {len(rag_graph.edges)}"
combined_graph_dump = ingestion_graph_copy.dump(
name="Combined Graph", description="Graph for data ingestion and RAG", endpoint_name="combined"