mypy

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.