Update depecrated fillna statements

DrafProject · May 22, 2024 · fd872f1 · fd872f1
1 parent 0c1a67b
commit fd872f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/elmada/from_entsoe.py b/elmada/from_entsoe.py
@@ -164,7 +164,7 @@ def load_el_national_generation(
 
     if fillna:
         df = fill_special_missing_data_points_for_gen(df=df, country=country, year=year)
-        df = df.fillna().ffill().fillna().bfill()
+        df = df.ffill().bfill()
 
     if resample:
         df = hp.resample(df, year=year, start_freq=data_freq, target_freq=freq)
@@ -373,7 +373,7 @@ def prep_dayahead_prices(
 
     if ensure_std_index:
         idx = hp.make_datetimeindex(year, hp.estimate_freq(ser), tz=ser.index.tz)
-        ser = ser.reindex(idx).fillna(method="ffill").fillna(method="bfill")
+        ser = ser.reindex(idx).ffill().bfill()
         ser = ser.reset_index(drop=True)
 
     if fillna:

diff --git a/elmada/helper.py b/elmada/helper.py
@@ -368,7 +368,7 @@ def fill_outlier_and_nan(
     if method == "linear":
         df = df.interpolate(method="linear")
     elif method == "fill":
-        df = df.fillna(method="bfill").fillna(method="ffill")
+        df = df.bfill().ffill()
 
     return df