Skip to content

Commit

Permalink
Fixed input gathering for collectors (#2818)
Browse files Browse the repository at this point in the history
  • Loading branch information
RunDevelopment authored and joeyballentine committed Apr 24, 2024
1 parent c071afd commit ef752e2
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions backend/src/process.py
Original file line number Diff line number Diff line change
Expand Up @@ -505,18 +505,20 @@ async def __gather_inputs(self, node: Node) -> list[object]:
assigned_inputs = self.inputs.get(node.id)
assert len(assigned_inputs) == len(node.data.inputs)

inputs = []
for input_index, node_input in enumerate(assigned_inputs):
async def get_input_value(input_index: int, node_input: Input):
if input_index in ignore:
inputs.append(None)
return None

if input_index in lazy:
inputs.append(
Lazy.from_coroutine(
self.__resolve_node_input(node_input), self.loop
)
return Lazy.from_coroutine(
self.__resolve_node_input(assigned_inputs[input_index]), self.loop
)
else:
inputs.append(await self.__resolve_node_input(node_input))

return await self.__resolve_node_input(node_input)

inputs = []
for input_index, node_input in enumerate(assigned_inputs):
inputs.append(await get_input_value(input_index, node_input))

return inputs

Expand Down

0 comments on commit ef752e2

Please sign in to comment.