From 801c23f418e14ae39a92646a41d6cb0d6b932505 Mon Sep 17 00:00:00 2001 From: Casper Guo Date: Fri, 17 Jan 2025 14:56:36 -0500 Subject: [PATCH] Fix setting timedelta on datetime array See #674 for discussion --- fastf1/core.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/fastf1/core.py b/fastf1/core.py index 9181cfee..e619c2f0 100644 --- a/fastf1/core.py +++ b/fastf1/core.py @@ -1551,7 +1551,7 @@ def _load_laps_data(self, livedata=None): laps_start_time.insert(0, self.session_start_time) else: laps_start_time.insert(0, pd.NaT) - laps_start_time = pd.Series(laps_start_time) + laps_start_time = pd.Series(laps_start_time, dtype="timedelta64[ns]") # don't set lap start times after red flag restart to the time # at which the previous lap was set @@ -1590,9 +1590,7 @@ def _load_laps_data(self, livedata=None): elif row['Status'] == 'Aborted': # red flag _is_aborted = True - result.loc[:, 'LapStartTime'] = pd.Series( - laps_start_time, dtype='timedelta64[ns]' - ) + result['LapStartTime'] = laps_start_time # set missing lap start times to pit out time, where possible mask = (pd.isna(result['LapStartTime'])