Filter NaN's when fitting JS

NannyML · Nov 19, 2023 · 369efbf · 369efbf
1 parent 0cfcf92
commit 369efbf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: