diff options
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
|