From daa30eb8ac12affe5f6d677d001abf1b77c48b71 Mon Sep 17 00:00:00 2001 From: Santiago Maass Gangler Date: Tue, 7 Jan 2025 10:37:38 -0300 Subject: [PATCH] FIX: n bins parameter utilization in psi stattest --- src/evidently/calculations/stattests/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/evidently/calculations/stattests/utils.py b/src/evidently/calculations/stattests/utils.py index ebd6f8d1f6..a8996d5e29 100644 --- a/src/evidently/calculations/stattests/utils.py +++ b/src/evidently/calculations/stattests/utils.py @@ -27,7 +27,7 @@ def get_binned_data( n_vals = reference_data.nunique() if feature_type == ColumnType.Numerical and n_vals > 20: - bins = np.histogram_bin_edges(pd.concat([reference_data, current_data], axis=0).values, bins="sturges") + bins = np.histogram_bin_edges(pd.concat([reference_data, current_data], axis=0).values, bins=n) reference_percents = np.histogram(reference_data, bins)[0] / len(reference_data) current_percents = np.histogram(current_data, bins)[0] / len(current_data)