diff options
| author | Ganesh Kathiresan <ganesh3597@gmail.com> | 2022-03-25 22:53:32 +0530 |
|---|---|---|
| committer | Ganesh Kathiresan <ganesh3597@gmail.com> | 2022-03-25 22:53:32 +0530 |
| commit | 2d2e7b592326f61ee78fb9b420d550a3a789cc4c (patch) | |
| tree | dec3ab26ecdab32dce0911983110efc0b73addd1 /numpy/lib | |
| parent | 54605f3f828c76639cc6dff5165c59ae2074988e (diff) | |
| download | numpy-2d2e7b592326f61ee78fb9b420d550a3a789cc4c.tar.gz | |
MAINT: Linting fixes
Diffstat (limited to 'numpy/lib')
| -rw-r--r-- | numpy/lib/shape_base.py | 2 | ||||
| -rw-r--r-- | numpy/lib/tests/test_shape_base.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py index 25a96b966..b973afb0e 100644 --- a/numpy/lib/shape_base.py +++ b/numpy/lib/shape_base.py @@ -1163,7 +1163,7 @@ def kron(a, b): # Reshape back result = result.reshape(as_+bs) - transposer = _nx.arange(nd*2).reshape([2,nd]).transpose().reshape(nd*2) + transposer = _nx.arange(nd*2).reshape([2, nd]).transpose().reshape(nd*2) result = result.transpose(transposer) result = result.reshape(_nx.multiply(as_, bs)) diff --git a/numpy/lib/tests/test_shape_base.py b/numpy/lib/tests/test_shape_base.py index d75a1006c..564cdfeea 100644 --- a/numpy/lib/tests/test_shape_base.py +++ b/numpy/lib/tests/test_shape_base.py @@ -679,7 +679,8 @@ class TestKron: expected_shape = np.multiply(normalised_shape_a, normalised_shape_b) k = np.kron(a, b) - assert np.array_equal(k.shape, expected_shape), "Unexpected shape from kron" + assert np.array_equal( + k.shape, expected_shape), "Unexpected shape from kron" class TestTile: |