Fix bug due to pandas release (#359)

* Change none no nan * Convert all data to np.nan
sdv-dev · Jan 26, 2022 · adc2b1a · adc2b1a
1 parent 40d2667
commit adc2b1a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rdt/transformers/categorical.py b/rdt/transformers/categorical.py
@@ -89,6 +89,7 @@ def _get_intervals(data):
             dict:
                 intervals for each categorical value (start, end).
         """
+        data = data.fillna(np.nan)
         frequencies = data.value_counts(dropna=False)
 
         start = 0