feat: Bump ruff version to 0.9 (#5666)
* Bump ruff version to 0.9 * Rename some modules for A005 ruff rule
This commit is contained in:
parent
242165b8aa
commit
da83dbbcb5
29 changed files with 137 additions and 136 deletions
|
|
@ -145,7 +145,7 @@ def test_vector_store_rag_dump_components_and_edges(ingestion_graph, rag_graph):
|
|||
for node_id, expected_type in expected_nodes.items():
|
||||
assert node_id in node_map, f"Missing node {node_id}"
|
||||
assert node_map[node_id]["type"] == expected_type, (
|
||||
f"Node {node_id} has incorrect type. " f"Expected {expected_type}, got {node_map[node_id]['type']}"
|
||||
f"Node {node_id} has incorrect type. Expected {expected_type}, got {node_map[node_id]['type']}"
|
||||
)
|
||||
|
||||
# Verify all nodes in graph are expected
|
||||
|
|
@ -226,9 +226,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)} != {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"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue