diff options
Diffstat (limited to 'ot/lp/solver_1d.py')
| -rw-r--r-- | ot/lp/solver_1d.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/lp/solver_1d.py b/ot/lp/solver_1d.py index e7add89..bcfc920 100644 --- a/ot/lp/solver_1d.py +++ b/ot/lp/solver_1d.py @@ -129,7 +129,7 @@ def wasserstein_1d(u_values, v_values, u_weights=None, v_weights=None, p=1, requ diff_quantiles = nx.abs(u_quantiles - v_quantiles) if p == 1: - return nx.sum(delta * nx.abs(diff_quantiles), axis=0) + return nx.sum(delta * diff_quantiles, axis=0) return nx.sum(delta * nx.power(diff_quantiles, p), axis=0) |