summaryrefslogtreecommitdiff
path: root/numpy/lib/function_base.py
diff options
context:
space:
mode:
authorDuke Vijitbenjaronk <warut.vijitbenjaronk@gmail.com>2017-03-07 11:11:53 -0600
committerEric Wieser <wieser.eric@gmail.com>2017-03-07 20:49:33 +0000
commit1588ae39ffb51ea916f03510671aab711fdfb568 (patch)
treec08267636b61856aa8d2726a0ac0bb92a327c96d /numpy/lib/function_base.py
parentdf9e134d1e40de4d3865597a239c5a1d68624f5d (diff)
downloadnumpy-1588ae39ffb51ea916f03510671aab711fdfb568.tar.gz
BUG: Fix np.average with object array weights
Fixes #8696
Diffstat (limited to 'numpy/lib/function_base.py')
-rw-r--r--numpy/lib/function_base.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py
index c54512c21..0903790bd 100644
--- a/numpy/lib/function_base.py
+++ b/numpy/lib/function_base.py
@@ -1135,7 +1135,7 @@ def average(a, axis=None, weights=None, returned=False):
wgt = wgt.swapaxes(-1, axis)
scl = wgt.sum(axis=axis, dtype=result_dtype)
- if (scl == 0.0).any():
+ if np.any(scl == 0.0):
raise ZeroDivisionError(
"Weights sum to zero, can't be normalized")