Skip to content

Commit 63c7450

Browse files
committed
fix test 3
1 parent 4061890 commit 63c7450

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

rdt/transformers/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,7 @@ def _fill_nan_with_none_series(data):
183183

184184
return data.fillna(sentinel).replace({sentinel: None})
185185

186+
186187
def fill_nan_with_none(data):
187188
"""Replace all nan values with None.
188189

tests/unit/transformers/test_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,9 @@ def test__fill_nan_with_none_series():
184184
expected_result_categorical = pd.Series(
185185
pd.Categorical(['a', 'b', 'c', 'd', None], categories=['a', 'b', 'c', 'd'])
186186
)
187-
pd.testing.assert_series_equal(result_categorical, expected_result_categorical)
187+
pd.testing.assert_series_equal(
188+
result_categorical, expected_result_categorical, check_dtype=False
189+
)
188190

189191

190192
def test_fill_nan_with_none_series():

0 commit comments

Comments
 (0)