diff options
Diffstat (limited to 'numpy/core/shape_base.py')
-rw-r--r-- | numpy/core/shape_base.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/numpy/core/shape_base.py b/numpy/core/shape_base.py index 3259f3b1d..50c46f38e 100644 --- a/numpy/core/shape_base.py +++ b/numpy/core/shape_base.py @@ -48,9 +48,9 @@ def atleast_1d(*arys): res = [] for ary in arys: ary = asanyarray(ary) - if len(ary.shape) == 0 : + if len(ary.shape) == 0: result = ary.reshape(1) - else : + else: result = ary res.append(result) if len(res) == 1: @@ -98,11 +98,11 @@ def atleast_2d(*arys): res = [] for ary in arys: ary = asanyarray(ary) - if len(ary.shape) == 0 : + if len(ary.shape) == 0: result = ary.reshape(1, 1) - elif len(ary.shape) == 1 : + elif len(ary.shape) == 1: result = ary[newaxis,:] - else : + else: result = ary res.append(result) if len(res) == 1: |