fix plot (#1332)

* fix plot * black
facebookresearch · Jan 13, 2022 · c1e83e2 · c1e83e2
1 parent 666e009
commit c1e83e2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nevergrad/benchmark/plotting.py b/nevergrad/benchmark/plotting.py
@@ -502,7 +502,11 @@ def __init__(
             # confidence lines
             for conf in self._get_confidence_arrays(vals, log=logplot):
                 plt.plot(vals[xaxis], conf, name_style[optim_name], label=optim_name, alpha=0.1)
-            text = "{} ({:.3g} <{:.3g}>)".format(optim_name, vals["loss"][-1], vals["loss"][-2])
+            text = "{} ({:.3g} <{:.3g}>)".format(
+                optim_name,
+                vals["loss"][-1],
+                vals["loss"][-2] if len(vals["loss"]) > 2 else float("nan"),
+            )
             if vals[xaxis].size:
                 legend_infos.append(LegendInfo(vals[xaxis][-1], vals["loss"][-1], line, text))
         if not (np.isnan(upperbound) or np.isinf(upperbound)):