PR number, authot update, and small backend fix

Author: eloitanguy

.coveragerc

@@ -0,0 +1,8 @@
+[run]
+omit =
+    /tmp/*
+    */_remote_module_non_scriptable.py
+    */site-packages/*
+
+[report]
+skip_covered = True

RELEASES.md

@@ -4,7 +4,7 @@
 
 #### New features
 
-- Added Sliced OT plans (PR #757)
+- Added Sliced OT plans (PR #767)
 
 ## 0.9.6.post1
 

coverage_help.md

@@ -0,0 +1,4 @@
+Example:
+
+    coverage run -m pytest test/test_ot.py
+    coverage html --rcfile=.coveragerc

debug.py

@@ -0,0 +1,32 @@
+# %%
+import numpy as np
+from ot.backend import get_backend
+import torch
+import ot
+from torch.optim import Adam
+
+
+# %%
+rng = np.random.RandomState(0)
+n = 10
+d = 2
+X = rng.randn(n, d)
+Y = rng.randn(n, d) + np.array([5.0, 0.0])[None, :]
+n_proj = 20
+P = ot.sliced.get_random_projections(d, n_proj)
+a = rng.uniform(0, 1, n)
+a /= a.sum()
+b = rng.uniform(0, 1, n)
+b /= b.sum()
+sw2 = ot.sliced.sliced_wasserstein_distance(X, Y, a=a, b=b, projections=P)
+
+# %%
+nx = get_backend(torch.tensor([0.0]))
+X_t = nx.from_numpy(X)
+Y_t = nx.from_numpy(Y)
+a_t = nx.from_numpy(a)
+b_t = nx.from_numpy(b)
+P_t = nx.from_numpy(P)
+sw2_t = ot.sliced.sliced_wasserstein_distance(X_t, Y_t, a=a_t, b=b_t, projections=P_t)
+
+# %%

ot/sliced.py

@@ -6,6 +6,7 @@
 #         Nicolas Courty   <ncourty@irisa.fr>
 #         Rémi Flamary <remi.flamary@polytechnique.edu>
 #         Eloi Tanguy <eloi.tanguy@math.cnrs.fr>
+#         Laetitia Chapel <laetitia.chapel@irisa.fr>
 #
 # License: MIT License
 
@@ -776,7 +777,7 @@ def sliced_plans(
     do_draw_thetas = thetas is None
     if do_draw_thetas:  # create thetas (n_proj, d)
         assert n_proj is not None, "n_proj must be specified if thetas is None"
-        thetas = get_random_projections(d, n_proj, backend=nx).T
+        thetas = get_random_projections(d, n_proj, backend=nx, type_as=X).T
 
         if warm_theta is not None:
             thetas = nx.concatenate([thetas, warm_theta[:, None].T], axis=0)

test/test_sliced.py

@@ -3,6 +3,7 @@
 # Author: Adrien Corenflos <adrien.corenflos@aalto.fi>
 #         Nicolas Courty <ncourty@irisa.fr>
 #         Eloi Tanguy <eloi.tanguy@math.cnrs.fr>
+#         Laetitia Chapel <laetitia.chapel@irisa.fr>
 #
 # License: MIT License