Fixed axes_grid cax based on cbar_edge

inclement · inclement · commit 9ea98f4b1ee8 · 2015-02-06T23:30:49.000Z
diff --git a/lib/mpl_toolkits/axes_grid1/axes_grid.py b/lib/mpl_toolkits/axes_grid1/axes_grid.py
@@ -643,7 +643,7 @@ def __init__(self, fig,
             elif self._colorbar_mode == "edge":
                 for index, ax in enumerate(self.axes_all):
                     col, row = self._get_col_row(index)
-                    if self._direction == "row":
+                    if self._colorbar_location in ("left", "right"):
                         ax.cax = self.cbar_axes[row]
                     else:
                         ax.cax = self.cbar_axes[col]
