diff options
| author | ncassereau-idris <84033440+ncassereau-idris@users.noreply.github.com> | 2021-11-02 13:42:02 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-02 13:42:02 +0100 |
| commit | a335324d008e8982be61d7ace937815a2bfa98f9 (patch) | |
| tree | 83c7f637597f10f6f3d20b15532e53fc65b51f22 /test/test_bregman.py | |
| parent | 0cb2b2efe901ed74c614046d250518769f870313 (diff) | |
[MRG] Backend for gromov (#294)
* bregman: small correction
* gromov backend first draft
* Removing decorators
* Reworked casting method
* Bug solve
* Removing casting
* Bug solve
* toarray renamed todense ; expand_dims removed
* Warning (jax not supporting sparse matrix) moved
* Mistake corrected
* test backend
* Sparsity test for older versions of pytorch
* Trying pytorch/1.10
* Attempt to correct torch sparse bug
* Backend version of gromov tests
* Random state introduced for remaining gromov functions
* review changes
* code coverage
* Docs (first draft, to be continued)
* Gromov docs
* Prettified docs
* mistake corrected in the docs
* little change
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'test/test_bregman.py')
| -rw-r--r-- | test/test_bregman.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index c1120ba..6923d31 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -477,8 +477,8 @@ def test_lazy_empirical_sinkhorn(nx): b = ot.unif(n) numIterMax = 1000 - X_s = np.reshape(np.arange(n), (n, 1)) - X_t = np.reshape(np.arange(0, n), (n, 1)) + X_s = np.reshape(np.arange(n, dtype=np.float64), (n, 1)) + X_t = np.reshape(np.arange(0, n, dtype=np.float64), (n, 1)) M = ot.dist(X_s, X_t) M_m = ot.dist(X_s, X_t, metric='euclidean') |