jax .data fix + backend typing fix in sliced_plans test function

Author: eloitanguy

ot/sliced.py

@@ -880,15 +880,15 @@ def sliced_plans(
                             )
                             ** (1 / p)
                         )
-                        * plan[k].data
+                        * plan[k]
                     )
                     for k in range(n_proj)
                 ]
             elif metric == "sqeuclidean":
                 costs = [
                     nx.sum(
                         (nx.sum((X[plan[k].row] - Y[plan[k].col]) ** 2, axis=1))
-                        * plan[k].data
+                        * plan[k]
                     )
                     for k in range(n_proj)
                 ]

test/test_sliced.py

@@ -925,7 +925,9 @@ def test_sliced_plans_backends(nx):
 
     x_b, y_b, a_b, b_b = nx.from_numpy(x, y, a, b)
 
-    thetas_b = ot.sliced.get_random_projections(d, n_proj, seed=0, backend=nx).T
+    thetas_b = ot.sliced.get_random_projections(
+        d, n_proj, seed=0, backend=nx, type_as=x_b
+    ).T
     thetas = nx.to_numpy(thetas_b)
 
     context = (