FIX: initialize _in_draw state as early as possible

tacaswell · tacaswell · commit 73b28cfecb94 · 2021-03-30T21:52:50.000-04:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -522,6 +522,7 @@ def __init__(self, *args, zs=0, zdir='z', depthshade=True, **kwargs):
         This is typically desired in scatter plots.
         """
         self._depthshade = depthshade
+        self._in_draw = False
         super().__init__(*args, **kwargs)
         self.set_3d_properties(zs, zdir)
 
@@ -548,7 +549,6 @@ def set_3d_properties(self, zs, zdir):
         # grab the current sizes as-is
         # In the base draw methods we access the attributes directly which
         # means we can not buffer the shuffling on the way out.
-        self._in_draw = False
         self._sizes3d = self._sizes
         self._linewidths3d = self._linewidths
         xs, ys, zs = self._offsets3d
@@ -666,6 +666,7 @@ def patch_collection_2d_to_3d(col, zs=0, zdir='z', depthshade=True):
     elif isinstance(col, PatchCollection):
         col.__class__ = Patch3DCollection
     col._depthshade = depthshade
+    col._in_draw = False
     col.set_3d_properties(zs, zdir)
 
 
