From 8e6210757881027b40bfc174974daa3241fb07ce Mon Sep 17 00:00:00 2001 From: Abhinav Jangda Date: Mon, 3 Feb 2025 20:25:09 +0000 Subject: [PATCH] fix for linter --- .../operators/kronecker_product_linear_operator.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/linear_operator/operators/kronecker_product_linear_operator.py b/linear_operator/operators/kronecker_product_linear_operator.py index db9b1935..1ae2047b 100644 --- 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