Skip to content

Commit 5484b9f

Browse files
committed
BUG: improved fix for np 1.9 errors
1 parent b163b1e commit 5484b9f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/matplotlib/tri/triinterpolate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ def _interpolate_multikeys(self, x, y, tri_index=None,
210210
ret_loc = np.empty(size_ret, dtype=np.float64)
211211
ret_loc[~mask_in] = np.nan
212212
ret_loc[mask_in] = self._interpolate_single_key(
213-
return_key, valid_tri_index, valid_x, valid_y).ravel() * scale
213+
return_key, valid_tri_index, valid_x, valid_y) * scale
214214
ret += [np.ma.masked_invalid(ret_loc.reshape(sh_ret), copy=False)]
215215

216216
return ret
@@ -462,9 +462,9 @@ def _interpolate_single_key(self, return_key, tri_index, x, y):
462462
dzdx = self._ReferenceElement.get_function_derivatives(
463463
alpha, J, ecc, dof)
464464
if return_key == 'dzdx':
465-
return dzdx[:, 0]
465+
return dzdx[:, 0, 0]
466466
else:
467-
return dzdx[:, 1]
467+
return dzdx[:, 1, 0]
468468
else:
469469
raise ValueError("Invalid return_key: " + return_key)
470470

0 commit comments

Comments
 (0)