Fixed axes_grid cax based on cbar_edge

inclement · inclement · commit c345d9c14e25 · 2015-01-31T00:46:10.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]
