Merge pull request #8610 from eric-wieser/patch-2

dstansby · web-flow · commit c3e24ebfbdb1 · 2017-05-12T11:54:59.000+01:00
BUG: Correct invocation of `expand_dims`
diff --git a/lib/matplotlib/tri/triinterpolate.py b/lib/matplotlib/tri/triinterpolate.py
@@ -898,7 +898,7 @@ def get_bending_matrices(self, J, ecc):
         pts = self.gauss_pts
         for igauss in range(self.n_gauss):
             alpha = np.tile(pts[igauss, :], n).reshape(n, 3)
-            alpha = np.expand_dims(alpha, 3)
+            alpha = np.expand_dims(alpha, 2)
             weight = weights[igauss]
             d2Skdksi2 = self.get_d2Sidksij2(alpha, ecc)
             d2Skdx2 = _prod_vectorized(d2Skdksi2, H_rot)
