From 5cd02414e0705eeb9c278c22497821334d1b4402 Mon Sep 17 00:00:00 2001 From: Felix F Zimmermann Date: Mon, 25 Nov 2024 16:53:28 +0100 Subject: [PATCH] mypy --- src/mrpro/operators/Jacobian.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mrpro/operators/Jacobian.py b/src/mrpro/operators/Jacobian.py index 3a54780b..f5a12f47 100644 --- a/src/mrpro/operators/Jacobian.py +++ b/src/mrpro/operators/Jacobian.py @@ -1,9 +1,9 @@ """Jacobian.""" from collections.abc import Callable -from typing import Unpack import torch +from typing_extensions import Unpack from mrpro.operators.LinearOperator import LinearOperator from mrpro.operators.Operator import Operator @@ -46,7 +46,7 @@ def adjoint(self, *x: torch.Tensor) -> tuple[torch.Tensor, ...]: # type:ignore[ if self._vjp is None: self._f_x0, self._vjp = torch.func.vjp(self._operator, *self._x0) assert self._vjp is not None # noqa: S101 (hint for mypy) - return (self._vjp(x)[0],) + return (self._vjp(*x)[0],) def forward(self, *x: torch.Tensor) -> tuple[torch.Tensor, ...]: # type:ignore[override] """Apply the operator.