diff options
author | tlacombe <lacombe1993@gmail.com> | 2020-07-07 12:37:51 +0200 |
---|---|---|
committer | tlacombe <lacombe1993@gmail.com> | 2020-07-07 12:37:51 +0200 |
commit | 42a399c273fde7c76ec23d2993957fcbb492ee79 (patch) | |
tree | 068c92bb31530fbdc0aa2029a7221be807b59c68 /src/python/gudhi | |
parent | e0eba14109e02676825f8c24563872a5b49c6120 (diff) |
correction mistake in tests
Diffstat (limited to 'src/python/gudhi')
-rw-r--r-- | src/python/gudhi/wasserstein/wasserstein.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/python/gudhi/wasserstein/wasserstein.py b/src/python/gudhi/wasserstein/wasserstein.py index 009c1bf7..981bbf08 100644 --- a/src/python/gudhi/wasserstein/wasserstein.py +++ b/src/python/gudhi/wasserstein/wasserstein.py @@ -214,7 +214,7 @@ def wasserstein_distance(X, Y, matching=False, order=1., internal_p=np.inf, enab # Zeroth step: check compatibility of arguments if keep_essential_parts and enable_autodiff: import warnings - warnings.warn("enable_autodiff does not handle essential parts yet. These will be ignored in the following computations") + warnings.warn("enable_autodiff does not handle essential parts yet. keep_essential_parts set to False.") keep_essential_parts = False # First step: handle empty diagrams @@ -256,11 +256,11 @@ def wasserstein_distance(X, Y, matching=False, order=1., internal_p=np.inf, enab return np.inf, None else: return np.inf # avoid computing off-diagonal transport cost if essential parts do not match (saves time) - else: essential_cost = 0 essential_matching = None + # Extract off-diaognal points of the diagrams. X, Y = _offdiag(X), _offdiag(Y) n = len(X) m = len(Y) |