correct bug with sparse matrices

lchapel · lchapel · commit dba0dec5d346 · 2025-12-04T13:51:13.000+01:00
diff --git a/examples/sliced-wasserstein/plot_sliced_plans.py b/examples/sliced-wasserstein/plot_sliced_plans.py
@@ -42,8 +42,8 @@
 alpha = 0.3
 
 
-proj_X = X @ thetas.T
-proj_Y = Y @ thetas.T
+# proj_X = X @ thetas.T
+# proj_Y = Y @ thetas.T
 
 
 ##############################################################################
diff --git a/ot/sliced.py b/ot/sliced.py
@@ -873,15 +873,15 @@ def sliced_plans(
                             )
                             ** (1 / p)
                         )
-                        * plan[k]
+                        * plan[k].data
                     )
                     for k in range(n_proj)
                 ]
             else:  # metric == "sqeuclidean"
                 costs = [
                     nx.sum(
                         (nx.sum((X[plan[k].row] - Y[plan[k].col]) ** 2, axis=1))
-                        * plan[k]
+                        * plan[k].data
                     )
                     for k in range(n_proj)
                 ]

Original file line number	Diff line number	Diff line change
`@@ -873,15 +873,15 @@ def sliced_plans(`
`873`	`873`	`)`
`874`	`874`	`** (1 / p)`
`875`	`875`	`)`
`876`		`- * plan[k]`
	`876`	`+ * plan[k].data`
`877`	`877`	`)`
`878`	`878`	`for k in range(n_proj)`
`879`	`879`	`]`
`880`	`880`	`else: # metric == "sqeuclidean"`
`881`	`881`	`costs = [`
`882`	`882`	`nx.sum(`
`883`	`883`	`(nx.sum((X[plan[k].row] - Y[plan[k].col]) ** 2, axis=1))`
`884`		`- * plan[k]`
	`884`	`+ * plan[k].data`
`885`	`885`	`)`
`886`	`886`	`for k in range(n_proj)`
`887`	`887`	`]`