fix history plot

parklab · Dec 21, 2023 · 03fd523 · 03fd523
1 parent a7ffc63
commit 03fd523
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/salamander/plot.py b/src/salamander/plot.py
@@ -50,8 +50,8 @@ def history_plot(values, conv_test_freq, min_iteration=0, ax=None, **kwargs):
         )
     if ax is None:
         _, ax = plt.subplots(figsize=(4, 4))
-        ax.set(xlabel="n_iteration", ylabel="objective function value")
 
+    ax.set(xlabel="n_iteration", ylabel="objective function value")
     min_index = next(
         idx
         for idx, n_iteration in enumerate(ns_iteration)