Draichi commited on
Commit
f45e0b6
·
unverified ·
1 Parent(s): 846ef4a

fix(formula1_databases): Telemetry datetime

Browse files
Files changed (1) hide show
  1. notebooks/formula1_databases.py +6 -1
notebooks/formula1_databases.py CHANGED
@@ -140,6 +140,9 @@ class FastF1ToSQL:
140
  # Load session data
141
  session.load()
142
 
 
 
 
143
  # Insert data into tables
144
  self.insert_event(session)
145
  self.insert_session(session)
@@ -280,6 +283,8 @@ class FastF1ToSQL:
280
  for _, lap in laps_per_driver.iterrows():
281
  lap_number = lap['LapNumber']
282
  telemetry = lap.get_telemetry()
 
 
283
 
284
  for _, sample in telemetry.iterrows():
285
  telemetry_data: dict[str, Any] = {
@@ -294,7 +299,7 @@ class FastF1ToSQL:
294
  'y_position': round(sample['Y'], 2),
295
  'z_position': round(sample['Z'], 2),
296
  'is_off_track': sample['Status'] == 'OffTrack',
297
- 'datetime': str(sample.name),
298
  }
299
  telemetry_data_list.append(telemetry_data)
300
 
 
140
  # Load session data
141
  session.load()
142
 
143
+ # Save session start date
144
+ self._session_start_date = session.session_info['StartDate']
145
+
146
  # Insert data into tables
147
  self.insert_event(session)
148
  self.insert_session(session)
 
283
  for _, lap in laps_per_driver.iterrows():
284
  lap_number = lap['LapNumber']
285
  telemetry = lap.get_telemetry()
286
+ telemetry['datetime'] = self._session_start_date + \
287
+ telemetry['Time']
288
 
289
  for _, sample in telemetry.iterrows():
290
  telemetry_data: dict[str, Any] = {
 
299
  'y_position': round(sample['Y'], 2),
300
  'z_position': round(sample['Z'], 2),
301
  'is_off_track': sample['Status'] == 'OffTrack',
302
+ 'datetime': str(sample['datetime']),
303
  }
304
  telemetry_data_list.append(telemetry_data)
305