formatting dumps for fastest lap

* object_type should be lap not fastest_lap * objects should be a list of dicts, not a dict * lap_number should be number
harningle · Sep 26, 2024 · c8edcd2 · c8edcd2
1 parent bcaf915
commit c8edcd2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/models/lap.py b/models/lap.py
@@ -24,7 +24,7 @@ class QualiLap(BaseModel):
 
 
 class FastestLap(BaseModel):
-    lap_number: PositiveInt
+    number: PositiveInt
     fastest_lap_rank: PositiveInt
 
     model_config = ConfigDict(extra='forbid')
@@ -39,8 +39,8 @@ class LapData(BaseModel):
 
 
 class FastestLapData(BaseModel):
-    object_type: str = 'fastest_lap'
+    object_type: str = 'lap'
     foreign_keys: SessionEntry
-    objects: FastestLap
+    objects: list[FastestLap]
 
     model_config = ConfigDict(extra='forbid')
diff --git a/parse_race_fastest_laps.py b/parse_race_fastest_laps.py
@@ -87,10 +87,10 @@ def to_json(df: pd.DataFrame) -> list[dict]:
                 session=session_type,
                 car_number=x['driver_no']
             ),
-            objects=FastestLap(
-                lap_number=x['lap'],
+            objects=[FastestLap(
+                number=x['lap'],
                 fastest_lap_rank=x['rank']
-            )
+            )]
         ).model_dump(),
         axis=1
     )