Merge pull request #1 from siufai6/siufai6-fix-901

Update lib.py to fix kernc#901
siufai6 · Aug 4, 2023 · 1d733b9 · 1d733b9
2 parents 0ce24d8 + 43d4b98
commit 1d733b9
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)