Allow set_ticks to take a locator object, since this is valid for the ticks kwarg in __init__.

dopplershift · dopplershift · commit 921e87ce6d01 · 2010-07-02T18:45:17.000Z
svn path=/trunk/matplotlib/; revision=8486
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -299,7 +299,11 @@ def set_ticks(self, ticks, update_ticks=True):
         set tick locations. Tick locations are updated immediately unless update_ticks is
         *False*. To manually update the ticks, call *update_ticks* method explicitly.
         """
-        self.locator = ticker.FixedLocator(ticks, nbins=len(ticks))
+        if cbook.iterable(ticks):
+            self.locator = ticker.FixedLocator(ticks, nbins=len(ticks))
+        else:
+            self.locator = ticks
+
         if update_ticks:
             self.update_ticks()
 
@@ -842,5 +846,3 @@ def make_axes(parent, **kw):
     cax = fig.add_axes(pbcb)
     cax.set_aspect(aspect, anchor=anchor, adjustable='box')
     return cax, kw
-
-
