summaryrefslogtreecommitdiff
path: root/numpy/core/shape_base.py
diff options
context:
space:
mode:
authorRoman Yurchak <rth.yurchak@pm.me>2018-11-24 20:59:42 +0100
committerRoman Yurchak <rth.yurchak@pm.me>2018-11-24 20:59:42 +0100
commit09992482c93f1b9e28b7958a792e6b3b709834fa (patch)
treeb807248742d02c48403d77b076f53b71f22df707 /numpy/core/shape_base.py
parent983bbb5ed2495e034b437a0b58a69371d4dfed74 (diff)
downloadnumpy-09992482c93f1b9e28b7958a792e6b3b709834fa.tar.gz
Use set litterals
Diffstat (limited to 'numpy/core/shape_base.py')
-rw-r--r--numpy/core/shape_base.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py
index 6d234e527..a529d2ad7 100644
--- a/numpy/core/shape_base.py
+++ b/numpy/core/shape_base.py
@@ -410,7 +410,7 @@ def stack(arrays, axis=0, out=None):
if not arrays:
raise ValueError('need at least one array to stack')
- shapes = set(arr.shape for arr in arrays)
+ shapes = {arr.shape for arr in arrays}
if len(shapes) != 1:
raise ValueError('all input arrays must have the same shape')