summaryrefslogtreecommitdiff
path: root/ot/gromov.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-03-21 10:07:47 +0100
committerRémi Flamary <remi.flamary@gmail.com>2018-03-21 10:07:47 +0100
commit927395b40dae98bcf027b601b6df48a4318cfef2 (patch)
tree9ab932f0da81a2e16415554babe7e8ce3faf4f6c /ot/gromov.py
parentfc9923dea2706b65ffe15fc86428cd8b53b5feb1 (diff)
add externals for function signature
Diffstat (limited to 'ot/gromov.py')
-rw-r--r--ot/gromov.py2
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