Skip to content

Commit 0d9c914

Browse files
fix: update numpy array comparisons to use isin (#2504)
Signed-off-by: Emmanuel Ferdman <[email protected]>
1 parent 998f833 commit 0d9c914

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

flopy/utils/zonbud.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -1010,12 +1010,12 @@ def _compute_mass_balance(self, kstpkper, totim):
10101010
rowidx = np.asarray(
10111011
(self._budget["time_step"] == kstpkper[0])
10121012
& (self._budget["stress_period"] == kstpkper[1])
1013-
& np.in1d(self._budget["name"], innames)
1013+
& np.isin(self._budget["name"], innames)
10141014
).nonzero()
10151015
elif totim is not None:
10161016
rowidx = np.asarray(
10171017
(self._budget["totim"] == totim)
1018-
& np.in1d(self._budget["name"], innames)
1018+
& np.isin(self._budget["name"], innames)
10191019
).nonzero()
10201020
a = _numpyvoid2numeric(self._budget[list(self._zonenamedict.values())][rowidx])
10211021
intot = np.array(a.sum(axis=0))
@@ -1028,12 +1028,12 @@ def _compute_mass_balance(self, kstpkper, totim):
10281028
rowidx = np.asarray(
10291029
(self._budget["time_step"] == kstpkper[0])
10301030
& (self._budget["stress_period"] == kstpkper[1])
1031-
& np.in1d(self._budget["name"], outnames)
1031+
& np.isin(self._budget["name"], outnames)
10321032
).nonzero()
10331033
elif totim is not None:
10341034
rowidx = np.asarray(
10351035
(self._budget["totim"] == totim)
1036-
& np.in1d(self._budget["name"], outnames)
1036+
& np.isin(self._budget["name"], outnames)
10371037
).nonzero()
10381038
a = _numpyvoid2numeric(self._budget[list(self._zonenamedict.values())][rowidx])
10391039
outot = np.array(a.sum(axis=0))

0 commit comments

Comments
 (0)