fmt, fix
This commit is contained in:
@@ -77,8 +77,7 @@ def resume_workflow(
|
||||
return run
|
||||
|
||||
while True:
|
||||
frame = select_next_frame(run)
|
||||
if frame is None:
|
||||
if select_next_frame(run) is None:
|
||||
status = resolve_no_ready_frames(run)
|
||||
if status == RunStatus.COMPLETED:
|
||||
break
|
||||
@@ -119,8 +118,7 @@ async def resume_workflow_async(
|
||||
return run
|
||||
|
||||
while True:
|
||||
frame = select_next_frame(run)
|
||||
if frame is None:
|
||||
if select_next_frame(run) is None:
|
||||
status = resolve_no_ready_frames(run)
|
||||
if status == RunStatus.COMPLETED:
|
||||
break
|
||||
|
||||
+12
-10
@@ -241,17 +241,19 @@ async def _step_async_foreach_item_batch(
|
||||
deterministically.
|
||||
"""
|
||||
frames = [first_frame, *_claim_matching_async_item_frames(run, index, first_frame)]
|
||||
tasks = [
|
||||
invoke_node_use_async_for_frame(
|
||||
workflow,
|
||||
run,
|
||||
frame,
|
||||
_node_use_for_frame(index, frame),
|
||||
index.node_defs[_node_use_for_frame(index, frame).node],
|
||||
registry,
|
||||
tasks = []
|
||||
for frame in frames:
|
||||
node = _node_use_for_frame(index, frame)
|
||||
tasks.append(
|
||||
invoke_node_use_async_for_frame(
|
||||
workflow,
|
||||
run,
|
||||
frame,
|
||||
node,
|
||||
index.node_defs[node.node],
|
||||
registry,
|
||||
)
|
||||
)
|
||||
for frame in frames
|
||||
]
|
||||
results = await asyncio.gather(*tasks, return_exceptions=True)
|
||||
for frame, result in zip(frames, results, strict=True):
|
||||
run.current_frame_id = frame.id
|
||||
|
||||
Reference in New Issue
Block a user