Further tweaks re add_axes.

Phil Elson · Phil Elson · commit 7674dcdc0aa7 · 2012-02-06T18:14:24.000Z
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -8439,7 +8439,7 @@ def label_outer(self):
 
 _subplot_classes = {}
 def subplot_class_factory(axes_class=None):
-    # This makes a new class that inherits from SubclassBase and the
+    # This makes a new class that inherits from SubplotBase and the
     # given axes_class (which is assumed to be a subclass of Axes).
     # This is perhaps a little bit roundabout to make a new class on
     # the fly like this, but it means that a new Subplot class does
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -703,16 +703,15 @@ def add_axes(self, *args, **kwargs):
                 projection = 'polar'
 
             if isinstance(projection, basestring) or projection is None:
-                projection_class = get_projection_class(projection)
+                a = projection_factory(projection, self, rect, **kwargs)
             elif hasattr(projection, '_as_mpl_axes'):
                 projection_class, extra_kwargs = projection._as_mpl_axes()
                 kwargs.update(**extra_kwargs)
+                a = projection_class(self, rect, **kwargs)                
             else:
                 TypeError('projection must be a string, None or implement a ' 
                           '_as_mpl_axes method. Got %r' % projection)
             
-            a = projection_factory(projection, self, rect, **kwargs)
-            
         self._axstack.add(key, a)
         self.sca(a)
         return a
