diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-03-21 10:07:47 +0100 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-03-21 10:07:47 +0100 |
commit | 927395b40dae98bcf027b601b6df48a4318cfef2 (patch) | |
tree | 9ab932f0da81a2e16415554babe7e8ce3faf4f6c /ot/gromov.py | |
parent | fc9923dea2706b65ffe15fc86428cd8b53b5feb1 (diff) |
add externals for function signature
Diffstat (limited to 'ot/gromov.py')
-rw-r--r-- | ot/gromov.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gromov.py b/ot/gromov.py index 2a23873..e03fa5b 100644 --- a/ot/gromov.py +++ b/ot/gromov.py @@ -595,7 +595,7 @@ def entropic_gromov_wasserstein2(C1, C2, p, q, loss_fun, epsilon, gw, logv = entropic_gromov_wasserstein(
C1, C2, p, q, loss_fun, epsilon, max_iter, tol, verbose, log=True)
- log['T'] = gw
+ logv['T'] = gw
if log:
return logv['gw_dist'], logv
|