Fix bug and test

wesselb · Sep 4, 2021 · 9cea80d · 9cea80d
1 parent 8965ec6
commit 9cea80d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/matrix/ops/sum.py b/matrix/ops/sum.py
@@ -112,7 +112,7 @@ def sum(a: Woodbury, axis: Union[B.Int, None] = None):
     else:
         # We're summing over batch dimensions: align the batches first.
         diag, lr = align_batch(a.diag, a.lr)
-        axis = resolved_axis(a, axis)
+        axis = resolve_axis(a, axis)
         return B.sum(diag, axis=axis) + B.sum(lr, axis=axis)
 
 

diff --git a/tests/ops/test_shape.py b/tests/ops/test_shape.py
@@ -3,7 +3,6 @@
 from lab.shape import Shape
 
 from matrix.ops.shape import _drop_axis
-
 # noinspection PyUnresolvedReferences
 from ..util import (
     AssertDenseWarning,
@@ -88,4 +87,5 @@ def test_drop_axis():
 
 
 def test_shape_tb(shape, tb1):
-    check_un_op(shape, tb1, asserted_type=(tuple, Shape))
+    with AssertDenseWarning(["tiling", "concatenating"]):
+        check_un_op(shape, tb1, asserted_type=(tuple, Shape))