summaryrefslogtreecommitdiff
path: root/examples/unbalanced-partial/plot_UOT_barycenter_1D.py
diff options
context:
space:
mode:
authorGard Spreemann <gspr@nonempty.org>2023-06-14 16:52:13 +0200
committerGard Spreemann <gspr@nonempty.org>2023-06-14 16:52:13 +0200
commit2b51c7bfcf54d7e17ac7c2514f54408543cbe126 (patch)
treebaf00cc603ceabad00626259eec898e6747d016c /examples/unbalanced-partial/plot_UOT_barycenter_1D.py
parenta49f648b0b07737f7ef315fb83d8f78871780281 (diff)
parent96788a3fe5601e4c3f49b592aa0d9c034247862e (diff)
Merge branch 'dfsg/latest'
Diffstat (limited to 'examples/unbalanced-partial/plot_UOT_barycenter_1D.py')
-rw-r--r--examples/unbalanced-partial/plot_UOT_barycenter_1D.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/unbalanced-partial/plot_UOT_barycenter_1D.py b/examples/unbalanced-partial/plot_UOT_barycenter_1D.py
index 931798b..8d227c0 100644
--- a/examples/unbalanced-partial/plot_UOT_barycenter_1D.py
+++ b/examples/unbalanced-partial/plot_UOT_barycenter_1D.py
@@ -127,7 +127,7 @@ for i, z in enumerate(zs):
ys = B_l2[:, i]
verts.append(list(zip(x, ys)))
-ax = pl.gcf().gca(projection='3d')
+ax = pl.gcf().add_subplot(projection='3d')
poly = PolyCollection(verts, facecolors=[cmap(a) for a in weight_list])
poly.set_alpha(0.7)
@@ -149,7 +149,7 @@ for i, z in enumerate(zs):
ys = B_wass[:, i]
verts.append(list(zip(x, ys)))
-ax = pl.gcf().gca(projection='3d')
+ax = pl.gcf().add_subplot(projection='3d')
poly = PolyCollection(verts, facecolors=[cmap(a) for a in weight_list])
poly.set_alpha(0.7)