Fix next call in graphexecutionstate.next

This commit is contained in:
Brandon Rising 2023-07-24 17:45:05 -04:00
parent d090be60e8
commit 7073dc0d5d

View File

@ -882,7 +882,7 @@ class GraphExecutionState(BaseModel):
if self.batch_indices[index] > 0:
self.executed.clear()
self.batch_indices[index] -= 1
return next(self)
return self.next(self)
# If next is still none, there's no next node, return None
return next_node