Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 9, 2023
1 parent 2d814ed commit deb5d58
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/moscot/base/problems/compound_problem.py
Original file line number Diff line number Diff line change
Expand Up @@ -382,7 +382,9 @@ def _(
for _src, _tgt in [(src, tgt)] + rest:
problem = self.problems[_src, _tgt]
fun = problem.push if forward else problem.pull
res[_tgt if forward else _src] = current_mass = fun(current_mass, scale_by_marginals=scale_by_marginals, **kwargs)
res[_tgt if forward else _src] = current_mass = fun(
current_mass, scale_by_marginals=scale_by_marginals, **kwargs
)

return res if return_all else current_mass

Expand Down

0 comments on commit deb5d58

Please sign in to comment.