From fab5b943ef092f30e8ce369d58f76ee73356f11c Mon Sep 17 00:00:00 2001 From: Dominik Klein Date: Fri, 3 Feb 2023 08:00:07 +0100 Subject: [PATCH] adapt plot_convergence --- src/moscot/backends/ott/_output.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/moscot/backends/ott/_output.py b/src/moscot/backends/ott/_output.py index bffaf7f2e..17ff3b4da 100644 --- a/src/moscot/backends/ott/_output.py +++ b/src/moscot/backends/ott/_output.py @@ -77,7 +77,7 @@ def select_values(last_k: Optional[int] = None) -> Tuple[str, jnp.ndarray, jnp.n kind, values, xs = select_values(last_k) ax.plot(xs, values, **kwargs) - ax.set_xlabel("iteration") + ax.set_xlabel("iteration (logged)") ax.set_ylabel(kind) if title is None: title = "converged" if self.converged else "not converged" # type: ignore[attr-defined]