From c06e1813cc0c490c7d9bce57103dd13b8f14f58b Mon Sep 17 00:00:00 2001 From: mrava87 Date: Thu, 25 Jun 2026 22:11:01 +0100 Subject: [PATCH] fix: change call of on_end callback to pass output x of step --- pylops/optimization/basesolver.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pylops/optimization/basesolver.py b/pylops/optimization/basesolver.py index 6b4faafa..0d9c23d6 100644 --- a/pylops/optimization/basesolver.py +++ b/pylops/optimization/basesolver.py @@ -129,7 +129,9 @@ def wrapper(*args, **kwargs): self, kwargs.get("x0", None) ) else: - getattr(cb, f"on_{func.__name__}_end")(self, args[0]) + getattr(cb, f"on_{func.__name__}_end")( + self, ret[0] if isinstance(ret, tuple) else ret + ) return ret return wrapper