diff --git a/flopy/discretization/unstructuredgrid.py b/flopy/discretization/unstructuredgrid.py index a3ae57e0d7..fbb4d6f0b0 100644 --- a/flopy/discretization/unstructuredgrid.py +++ b/flopy/discretization/unstructuredgrid.py @@ -51,7 +51,7 @@ def __init__(self, vertices=None, iverts=None, xcenters=None, ycenters=None, else: msg = ('Length of iverts must equal ncpl.sum ' '({} {})'.format(len(iverts), ncpl)) - assert len(iverts) == ncpl.sum(), msg + assert len(iverts) == np.sum(ncpl), msg assert np.array(self.xcellcenters).shape[0] == self.ncpl assert np.array(self.ycellcenters).shape[0] == self.ncpl diff --git a/flopy/plot/map.py b/flopy/plot/map.py index 365b4b9e5b..dd4b4668bb 100644 --- a/flopy/plot/map.py +++ b/flopy/plot/map.py @@ -170,7 +170,7 @@ def plot_array(self, a, masked_values=None, **kwargs): else: quadmesh = plotutil.plot_cvfd(self.mg._vertices, self.mg._iverts, - a=a, ax=ax) + a=plotarray, ax=ax) # set max and min if 'vmin' in kwargs: