Fix BlendedAffine creating Affine2D in all cases

trananso · trananso · commit 5f6d8e355b17 · 2024-05-07T16:44:45.000-04:00
diff --git a/lib/matplotlib/transforms.py b/lib/matplotlib/transforms.py
@@ -2648,8 +2648,9 @@ def __init__(self, *args, **kwargs):
         `blended_transform_factory` function instead, which can determine
         automatically which kind of blended transform to create.
         """
+        dims = len(args)
         Transform.__init__(self, **kwargs)
-        AffineImmutable.__init__(self, **kwargs)
+        AffineImmutable.__init__(self, dims=dims, **kwargs)
 
         if not all(transform.is_affine and transform.is_separable
                    for transform in args):
@@ -2799,8 +2800,9 @@ def get_affine(self):
         if not self._b.is_affine:
             return self._b.get_affine()
         else:
-            return Affine2D(np.dot(self._b.get_affine().get_matrix(),
-                                   self._a.get_affine().get_matrix()))
+            return _affine_factory(np.dot(self._b.get_affine().get_matrix(),
+                                   self._a.get_affine().get_matrix()),
+                                   dims=self.input_dims)
 
     def inverted(self):
         # docstring inherited
