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

@ -116,7 +116,7 @@ class TestStructuredOutputComponent:
"display_name": "Type",
"type": "str",
"description": (
"Indicate the data type of the output field " "(e.g., str, int, float, bool, list, dict)."
"Indicate the data type of the output field (e.g., str, int, float, bool, list, dict)."
),
},
{
@ -158,7 +158,7 @@ class TestStructuredOutputComponent:
"display_name": "Type",
"type": "str",
"description": (
"Indicate the data type of the output field " "(e.g., str, int, float, bool, list, dict)."
"Indicate the data type of the output field (e.g., str, int, float, bool, list, dict)."
),
},
{

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"