Update functions/pitch_summary_functions.py

functions/pitch_summary_functions.py

@@ -1073,6 +1073,12 @@ def stat_summary_table(df: pl.DataFrame,
     split : str, optional
         The split type (default is 'All').
     """
+
+    type_dict =  {'R':'Regular Season',
+                   'S':'Spring',
+                   'P':'Playoffs' }
+
+    
     # Format start and end dates
     start_date_format = str(pd.to_datetime(df['game_date'][0]).strftime('%m/%d/%Y'))
     end_date_format = str(pd.to_datetime(df['game_date'][-1]).strftime('%m/%d/%Y'))
@@ -1107,10 +1113,10 @@ def stat_summary_table(df: pl.DataFrame,
         pitcher_stats_call_df_small = pitcher_stats_call_df.select(['inningsPitched', 'battersFaced', 'earnedRuns', 'hits', 'strikeOuts', 'baseOnBalls', 'hitByPitch', 'homeRuns', 'strikePercentage', 'whiffs'])
         new_column_names = ['$\\bf{IP}$', '$\\bf{PA}$', '$\\bf{ER}$', '$\\bf{H}$', '$\\bf{K}$', '$\\bf{BB}$', '$\\bf{HBP}$', '$\\bf{HR}$', '$\\bf{Strike\%}$', '$\\bf{Whiffs}$']
         title = f'{df["game_date"][0]} vs {df["batter_team"][0]}'
-    elif sport_id != 1:
+    elif sport_id != 1 or game_type in ['S','P']:
         pitcher_stats_call_df_small = pitcher_stats_call_df.select(['inningsPitched', 'battersFaced', 'whip', 'era', 'fip', 'k_percent', 'bb_percent', 'k_bb_percent', 'strikePercentage'])
         new_column_names = ['$\\bf{IP}$', '$\\bf{PA}$', '$\\bf{WHIP}$', '$\\bf{ERA}$', '$\\bf{FIP}$', '$\\bf{K\%}$', '$\\bf{BB\%}$', '$\\bf{K-BB\%}$', '$\\bf{Strike\%}$']
-        title = f'{df["game_date"][0]} to {df["game_date"][-1]}'
+        title = f'{type_dict[game_type[0]]} {df["game_date"][0]} to {df["game_date"][-1]}'
     else:
         fangraphs_table(df=df, ax=ax, player_input=player_input, season=int(df['game_date'][0][0:4]), split=split)
         return