adapt plot_convergence

theislab · Feb 3, 2023 · fab5b94 · fab5b94
1 parent 1f36cd1
commit fab5b94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]