fix for linter

cornellius-gp · Feb 3, 2025 · 8e62107 · 8e62107
1 parent 05d8eb6
commit 8e62107
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/linear_operator/operators/kronecker_product_linear_operator.py b/linear_operator/operators/kronecker_product_linear_operator.py
@@ -271,9 +271,10 @@ def _matmul(
     ) -> Union[Float[torch.Tensor, "... M C"], Float[torch.Tensor, "... M"]]:
         res = fktorch.gekmm([op.to_dense() for op in self.linear_ops], rhs.contiguous())
         return res
-
-    def rmatmul(self: Float[LinearOperator, "... M N"],
-                rhs: Union[Float[Tensor, "... P M"], Float[Tensor, "... M"], Float[LinearOperator, "... P M"]],
+
+    def rmatmul(
+        self: Float[LinearOperator, "... M N"],
+        rhs: Union[Float[Tensor, "... P M"], Float[Tensor, "... M"], Float[LinearOperator, "... P M"]],
     ) -> Union[Float[Tensor, "... P N"], Float[Tensor, "N"], Float[LinearOperator, "... P N"]]:
         res = fktorch.gemkm(rhs.contiguous(), [op.to_dense() for op in self.linear_ops])
         return res