fix: add vertices in the first layer to the vertices being run (#2594)

This commit is contained in:
Gabriel Luiz Freitas Almeida
2024-07-09 12:03:55 -03:00
committed by GitHub
parent 21adbd5531
commit c20630b77d

View File

@ -106,6 +106,9 @@ async def retrieve_vertices_order(
else:
first_layer = graph.sort_vertices()
for vertex_id in first_layer:
graph.run_manager.add_to_vertices_being_run(vertex_id)
# Now vertices is a list of lists
# We need to get the id of each vertex
# and return the same structure but only with the ids