diff options
-rw-r--r-- | ot/bregman.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index f6aa339..9e9989f 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1375,9 +1375,9 @@ def empirical_sinkhorn(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', numI ''' if a is None: - a = utils.unif(np.shape(X_s)[0]) + a = ot.unif(np.shape(X_s)[0]) if b is None: - b = utils.unif(np.shape(X_t)[0]) + b = ot.unif(np.shape(X_t)[0]) M = ot.dist(X_s, X_t, metric=metric) @@ -1465,9 +1465,9 @@ def empirical_sinkhorn2(X_s, X_t, reg, a=None, b=None, metric='sqeuclidean', num ''' if a is None: - a = utils.unif(np.shape(X_s)[0]) + a = ot.unif(np.shape(X_s)[0]) if b is None: - b = utils.unif(np.shape(X_t)[0]) + b = ot.unif(np.shape(X_t)[0]) M = ot.dist(X_s, X_t, metric=metric) |