figsize

examples/unbalanced-partial/plot_partial_1d.py

@@ -61,7 +61,7 @@ def plot_partial_transport(
 x_b = np.sort(np.random.uniform(0, 10, size=n))
 
 # Plot original distributions
-plt.figure(figsize=(10, 2))
+plt.figure(figsize=(6, 2))
 plot_partial_transport(plt.gca(), x_a, x_b)
 plt.show()
 
@@ -73,7 +73,7 @@ def plot_partial_transport(
 
 # Visualize all partial transport plans
 for k in range(n):
-    plt.figure(figsize=(10, 2))
+    plt.figure(figsize=(6, 2))
     plot_partial_transport(
         plt.gca(),
         x_a,