ref: Fix some ruff rules for private access (SLF) (#4139)

* Add ruff rules for private access (SLF)

* Changes following review

* Rename Vertex._data to Vertex.full_data
This commit is contained in:
Christophe Bornet 2024-10-23 01:04:12 +02:00 • committed by GitHub
commit f44aca5b41
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 199 additions and 199 deletions

View file

@ -169,4 +169,4 @@ async def run_single_component(
_, _ = await run_graph_internal(
graph, flow_id, session_id=session_id, inputs=graph_run_inputs, outputs=[component_id]
)
return graph.get_vertex(component_id)._built_object
return graph.get_vertex(component_id).built_object

View file

@ -126,13 +126,13 @@ def test_that_outputs_cache_is_set_to_false_in_cycle():
graph = Graph(chat_input, chat_output)
cycle_vertices = find_cycle_vertices(graph._get_edges_as_list_of_tuples())
cycle_outputs_lists = [graph.vertex_map[vertex_id]._custom_component.outputs for vertex_id in cycle_vertices]
cycle_outputs_lists = [graph.vertex_map[vertex_id].custom_component.outputs for vertex_id in cycle_vertices]
cycle_outputs = [output for outputs in cycle_outputs_lists for output in outputs]
for output in cycle_outputs:
assert output.cache is False
non_cycle_outputs_lists = [
vertex._custom_component.outputs for vertex in graph.vertices if vertex.id not in cycle_vertices
vertex.custom_component.outputs for vertex in graph.vertices if vertex.id not in cycle_vertices
]
non_cycle_outputs = [output for outputs in non_cycle_outputs_lists for output in outputs]
for output in non_cycle_outputs: