diff --git a/menelaus/data_drift/psi_detector.py b/menelaus/data_drift/psi_detector.py index 9ac653d..efad3a8 100644 --- a/menelaus/data_drift/psi_detector.py +++ b/menelaus/data_drift/psi_detector.py @@ -94,6 +94,7 @@ def update(self, X: np.array, y_true=None, y_pred=None): self._drift_state = "drift" self.set_reference(test_batch) return psi_value + def _bin_data(self, feature, min, max): """ Bin the given feature based on the specified minimum and maximum values.