Apply review comments

Author: mtsokol

Diff for: src/finch/linalg/_linalg.py

@@ -1,3 +1,5 @@
+from numpy.core.numeric import normalize_axis_tuple
+
 from ..julia import jl
 from ..tensor import Tensor
 
@@ -11,12 +13,14 @@ def vector_norm(
     ord: int | float = 2,
 ) -> Tensor:
     if axis is not None:
-        raise ValueError(
-            "At the moment only `None` (vector norm of a flattened array) "
-            "is supported. Got: {axis}."
-        )
+        axis = normalize_axis_tuple(axis, x.ndim)
+        if axis != tuple(range(x.ndim)):
+            raise ValueError(
+                "At the moment only `None` (vector norm of a flattened array) "
+                "is supported. Got: {axis}."
+            )
 
     result = Tensor(jl.Finch.norm(x._obj, ord))
     if keepdims:
-        result = result.__getitem__(tuple(None for _ in range(x.ndim)))
+        result = result[tuple(None for _ in range(x.ndim))]
     return result

Diff for: src/finch/tensor.py

@@ -298,7 +298,7 @@ def _order(self) -> tuple[int, ...]:
     @property
     def mT(self) -> "Tensor":
         axes = list(range(self.ndim))
-        axes[-2:] = axes[:-3:-1]
+        axes[-2], axes[-1] = axes[-1], axes[-2]
         axes = tuple(axes)
         return self.permute_dims(axes)