Simplify end_dts

openclimatefix · Aug 14, 2024 · bded8db · bded8db
1 parent 6d08e10
commit bded8db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ocf_data_sampler/select/fill_time_periods.py b/ocf_data_sampler/select/fill_time_periods.py
@@ -5,8 +5,8 @@
 
 
 def fill_time_periods(time_periods: pd.DataFrame, freq: pd.Timedelta):
-    start_dts = time_periods["start_dt"].apply(lambda x: pd.Timestamp(x).ceil(freq))
-    end_dts = time_periods['end_dt'].apply(lambda x: pd.Timestamp(x))
+    start_dts = pd.to_datetime(time_periods["start_dt"].values).ceil(freq)
+    end_dts = pd.to_datetime(time_periods['end_dt'])
     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))