Skip to content

Commit

Permalink
f* pep8
Browse files Browse the repository at this point in the history
  • Loading branch information
rflamary committed Oct 18, 2023
1 parent 242c10d commit c60b412
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions ot/solvers.py
Original file line number Diff line number Diff line change
Expand Up @@ -542,8 +542,8 @@ def solve_gromov(Ca, Cb, M=None, a=None, b=None, loss='L2', symmetric=None, alph
tol = 1e-9

plan, log = sinkhorn_log(a, b, M, reg=reg, numItermax=max_iter,
stopThr=tol, log=True,
verbose=verbose)
stopThr=tol, log=True,
verbose=verbose)

value_linear = nx.sum(M * plan)
value = value_linear + reg * nx.sum(plan * nx.log(plan + 1e-16))
Expand Down

0 comments on commit c60b412

Please sign in to comment.