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
commit c20630b77d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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