Update lib.py to fix kernc#901

Heatmap plots requires at least two parameters (variables) in the heatmap Series. Add an assert before calling _plot_heatmap to guard against the case of having only 1 variable.
siufai6 · Aug 3, 2023 · 43d4b98 · 43d4b98
1 parent 0ce24d8
commit 43d4b98
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backtesting/lib.py b/backtesting/lib.py
@@ -138,6 +138,7 @@ def plot_heatmaps(heatmap: pd.Series,
     [plot_objective]: \
         https://scikit-optimize.github.io/stable/modules/plots.html#plot-objective
     """
+    assert heatmap.index.nlevels >= 2, "Heatmap plots requires at least 2 optimized parameters."
     return _plot_heatmaps(heatmap, agg, ncols, filename, plot_width, open_browser)