From adc2b1a33250b1b3b107eb033e2a75db7909a85f Mon Sep 17 00:00:00 2001
From: Felipe Alex Hofmann <fealho@gmail.com>
Date: Tue, 25 Jan 2022 18:05:48 -0800
Subject: [PATCH] Fix bug due to `pandas` release (#359)

* Change none no nan

* Convert all data to np.nan
---
 rdt/transformers/categorical.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/rdt/transformers/categorical.py b/rdt/transformers/categorical.py
index 97bcf71fb..b0b0b1636 100644
--- 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