bugfix

rtavenar · rtavenar · commit 2ba8d56cf4f6 · 2025-07-02T20:59:19.000+02:00
diff --git a/examples/unbalanced-partial/plot_partial_1d.py b/examples/unbalanced-partial/plot_partial_1d.py
@@ -34,7 +34,7 @@ def plot_partial_transport(
     if marginal_costs is not None:
         k = len(marginal_costs)
         ax.set_title(
-            f"Partial Transport - k = {k}, Cumulative Cost = {marginal_costs.sum():.2f}"
+            f"Partial Transport - k = {k}, Cumulative Cost = {sum(marginal_costs):.2f}"
         )
     else:
         ax.set_title("Original 1D Discrete Distributions")

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ def plot_partial_transport(`
`34`	`34`	`if marginal_costs is not None:`
`35`	`35`	`k = len(marginal_costs)`
`36`	`36`	`ax.set_title(`
`37`		`- f"Partial Transport - k = {k}, Cumulative Cost = {marginal_costs.sum():.2f}"`
	`37`	`+ f"Partial Transport - k = {k}, Cumulative Cost = {sum(marginal_costs):.2f}"`
`38`	`38`	`)`
`39`	`39`	`else:`
`40`	`40`	`ax.set_title("Original 1D Discrete Distributions")`