Fix merge on game id and add additional column for grouping

src/pages/9_Team_Formations.py

@@ -35,6 +35,12 @@ def get_page():
     ]
     group_options = [
         "week",
+        "down",
+        "qtr",
+        "ydstogo",
+        "play_type",
+        "pass_length",
+        "pass_location",
         "possession_team",
         "offense_formation",
         "offense_personnel",

src/queries/nflverse/github_data.py

@@ -46,9 +46,16 @@ def get_pbp_participation(season_int: int) -> pd.DataFrame:
                     SELECT 
                         a.*
                         , b.week
+                        , b.down
+                        , b.qtr
+                        , b.ydstogo
+                        , b.play_type
+                        , b.pass_length
+                        , b.pass_location
                     from pbp_participation_pbp_participation_{season_int} a
                     left join pbp_play_by_play_{season_int} b
                     on a.play_id = b.play_id
+                    and a.nflverse_game_id = b.game_id
 """
     ).df()
     return df