Use consistent array shape when setting square axis.

QuLogic · QuLogic · commit df618987e646 · 2019-07-25T21:26:30.000-04:00
NumPy 1.18.x will not like setting an array slice with a ragged array
like this.
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -1645,7 +1645,7 @@ def axis(self, *args, emit=True, **kwargs):
                     self.set_autoscale_on(False)
                     xlim = self.get_xlim()
                     ylim = self.get_ylim()
-                    edge_size = max(np.diff(xlim), np.diff(ylim))
+                    edge_size = max(np.diff(xlim), np.diff(ylim))[0]
                     self.set_xlim([xlim[0], xlim[0] + edge_size],
                                   emit=emit, auto=False)
                     self.set_ylim([ylim[0], ylim[0] + edge_size],
