Skip to content

Commit e26d32f

Browse files
committed
fix: remove deprecated nan
1 parent dc154c6 commit e26d32f

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

pytoda/datasets/_csv_statistics.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Abstract implementation of _CsvStatistics."""
2+
23
import copy
34
from functools import reduce
45

@@ -36,13 +37,11 @@ def __init__(
3637
self.min_max_scaler = MinMaxScaler()
3738
self.standardizer = StandardScaler()
3839
self.kwargs = copy.deepcopy(kwargs)
39-
self.kwargs['dtype'] = pandas_dtype
40+
self.kwargs["dtype"] = pandas_dtype
4041

4142
self.preprocess_df = (
4243
# may be applied to many chunks, so logic is determined once here
43-
self._reindex
44-
if self.initial_feature_list
45-
else self._id
44+
self._reindex if self.initial_feature_list else self._id
4645
)
4746
self.setup_datasource()
4847

@@ -54,7 +53,7 @@ def __init__(
5453

5554
def _reindex(self, df: pd.DataFrame) -> pd.DataFrame:
5655
"""Ensure given order of features, creating NaN columns for missing."""
57-
return df.reindex(columns=self.initial_feature_list, fill_value=np.NaN)
56+
return df.reindex(columns=self.initial_feature_list, fill_value=np.nan)
5857

5958
def _id(self, df: pd.DataFrame) -> pd.DataFrame:
6059
return df

0 commit comments

Comments
 (0)