Make them look the same

openclimatefix · Aug 14, 2024 · 09d5e09 · 09d5e09
1 parent bded8db
commit 09d5e09
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ocf_data_sampler/select/fill_time_periods.py b/ocf_data_sampler/select/fill_time_periods.py
@@ -6,7 +6,6 @@
 
 def fill_time_periods(time_periods: pd.DataFrame, freq: pd.Timedelta):
     start_dts = pd.to_datetime(time_periods["start_dt"].values).ceil(freq)
-    end_dts = pd.to_datetime(time_periods['end_dt'])
+    end_dts = pd.to_datetime(time_periods["end_dt"].values)
     date_ranges = [pd.date_range(start_dt, end_dt, freq=freq) for start_dt, end_dt in zip(start_dts, end_dts)]
-
     return pd.DatetimeIndex(np.concatenate(date_ranges))