Return collection in add_collection3d()

thangleiter · thangleiter · commit d2b5261b5012 · 2020-07-07T18:20:07.000+02:00
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -2249,7 +2249,8 @@ def add_collection3d(self, col, zs=0, zdir='z'):
             art3d.patch_collection_2d_to_3d(col, zs=zs, zdir=zdir)
             col.set_sort_zpos(zsortval)
 
-        super().add_collection(col)
+        collection = super().add_collection(col)
+        return collection
 
     def scatter(self, xs, ys, zs=0, zdir='z', s=20, c=None, depthshade=True,
                 *args, **kwargs):
