Corrected cax attributes of ImageGrid axes

inclement · inclement · commit 695f90e3c841 · 2015-01-31T00:29:06.000Z
diff --git a/lib/mpl_toolkits/axes_grid1/axes_grid.py b/lib/mpl_toolkits/axes_grid1/axes_grid.py
@@ -640,6 +640,13 @@ def __init__(self, fig,
             if self._colorbar_mode == "single":
                 for ax in self.axes_all:
                     ax.cax = self.cbar_axes[0]
+            elif self._colorbar_mode == "edge":
+                for index, ax in enumerate(self.axes_all):
+                    col, row = self._get_col_row(index)
+                    if self._direction == "row":
+                        ax.cax = self.cbar_axes[row]
+                    else:
+                        ax.cax = self.cbar_axes[col]
             else:
                 for ax, cax in zip(self.axes_all, self.cbar_axes):
                     ax.cax = cax
