Fix "UnboundLocalErrors" in mlab.py as reported by @huard mdboom@d6ec786#commitcomment-1061634

mdboom · mdboom · commit b765d87be811 · 2012-03-08T11:59:45.000-05:00
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -2999,15 +2999,17 @@ def poly_below(xmin, xs, ys):
       ax.fill(xv, yv)
     """
     if ma.isMaskedArray(xs) or ma.isMaskedArray(ys):
-        np = ma
+        numpy = ma
+    else:
+        numpy = np
 
-    xs = np.asarray(xs)
-    ys = np.asarray(ys)
+    xs = numpy.asarray(xs)
+    ys = numpy.asarray(ys)
     Nx = len(xs)
     Ny = len(ys)
     assert(Nx==Ny)
-    x = xmin*np.ones(2*Nx)
-    y = np.ones(2*Nx)
+    x = xmin*numpy.ones(2*Nx)
+    y = numpy.ones(2*Nx)
     x[:Nx] = xs
     y[:Nx] = ys
     y[Nx:] = ys[::-1]
@@ -3026,17 +3028,19 @@ def poly_between(x, ylower, yupper):
     :meth:`matplotlib.axes.Axes.fill`.
     """
     if ma.isMaskedArray(ylower) or ma.isMaskedArray(yupper) or ma.isMaskedArray(x):
-        np = ma
+        numpy = ma
+    else:
+        numpy = np
 
     Nx = len(x)
     if not cbook.iterable(ylower):
-        ylower = ylower*np.ones(Nx)
+        ylower = ylower*numpy.ones(Nx)
 
     if not cbook.iterable(yupper):
-        yupper = yupper*np.ones(Nx)
+        yupper = yupper*numpy.ones(Nx)
 
-    x = np.concatenate( (x, x[::-1]) )
-    y = np.concatenate( (yupper, ylower[::-1]) )
+    x = numpy.concatenate( (x, x[::-1]) )
+    y = numpy.concatenate( (yupper, ylower[::-1]) )
     return x,y
 
 
