From 369efbfe9ec05d915e01a3d330e9764d0571eb0a Mon Sep 17 00:00:00 2001 From: Michael Van de Steene Date: Sun, 19 Nov 2023 19:24:34 +0100 Subject: [PATCH] Filter NaN's when fitting JS --- nannyml/drift/univariate/methods.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nannyml/drift/univariate/methods.py b/nannyml/drift/univariate/methods.py index 73e61a1a..899adec8 100644 --- a/nannyml/drift/univariate/methods.py +++ b/nannyml/drift/univariate/methods.py @@ -278,6 +278,7 @@ def __init__(self, **kwargs) -> None: self._reference_proba_in_bins: np.ndarray def _fit(self, reference_data: pd.Series, timestamps: Optional[pd.Series] = None): + reference_data, = _clean_data(reference_data) if _column_is_categorical(reference_data): treat_as_type = 'cat' else: