From 220c8617718a98f3b22ba9ea4f9f63a12a602672 Mon Sep 17 00:00:00 2001
From: EtienneRobert <100388686+EtienneRobert@users.noreply.github.com>
Date: Tue, 28 Feb 2023 14:11:43 -0500
Subject: [PATCH] Update ale.py

---
 src/alepython/ale.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/alepython/ale.py b/src/alepython/ale.py
index 819f0bb..aaaebc2 100644
--- a/src/alepython/ale.py
+++ b/src/alepython/ale.py
@@ -646,6 +646,7 @@ def ale_plot(
     monte_carlo_rep=50,
     monte_carlo_ratio=0.1,
     rugplot_lim=1000,
+    mark_empty=True
 ):
     """Plots ALE function of specified features based on training set.
 
@@ -677,6 +678,8 @@ def ale_plot(
     rugplot_lim : int, optional
         If `train_set` has more rows than `rugplot_lim`, no rug plot will be plotted.
         Set to None to always plot rug plots. Set to 0 to always plot rug plots.
+    mark_empty : bool, optional
+        If True, plot rectangles over bins that did not contain any samples. 
 
     Raises
     ------
@@ -764,7 +767,7 @@ def ale_plot(
                 features,
                 bins,
             )
-            _second_order_quant_plot(fig, ax, quantiles_list, ale)
+            _second_order_quant_plot(fig, ax, quantiles_list, ale, mark_empty=mark_empty)
             _ax_labels(
                 ax,
                 "Feature '{}'".format(features[0]),