Skip to content

Commit

Permalink
Feed inputs to get_forks so fork factory can use algo input data (#129)
Browse files Browse the repository at this point in the history
  • Loading branch information
george-zubrienko authored Nov 26, 2024
1 parent adfbb1e commit 0545cf1
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions esd_services_api_client/nexus/algorithms/forked_algorithm.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,8 +143,13 @@ async def _measured_run(**run_args) -> AlgorithmResult:

return await self._main_run(**run_args)

if self._is_forked(**kwargs):
self._inputs = await self._fork_inputs(**kwargs)
else:
self._inputs = await self._main_inputs(**kwargs)

# evaluate if additional forks will be spawned
forks = await self._get_forks(**kwargs)
forks = await partial(self._get_forks, **self._inputs, **kwargs)()

if len(forks) > 0:
self._logger.info(
Expand All @@ -154,11 +159,6 @@ async def _measured_run(**run_args) -> AlgorithmResult:
else:
self._logger.info("Leaf algorithm node: proceeding with this node run only")

if self._is_forked(**kwargs):
self._inputs = await self._fork_inputs(**kwargs)
else:
self._inputs = await self._main_inputs(**kwargs)

run_result = await partial(
_measured_run,
**self._inputs,
Expand Down

0 comments on commit 0545cf1

Please sign in to comment.